Merge pull request #159 from gjp4tw/main

fix typo
leondgarse · Apr 9, 2024 · 488ecac · 488ecac
2 parents a143ffb + 47c9c09
commit 488ecac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras_cv_attention_models/common_layers.py b/keras_cv_attention_models/common_layers.py
@@ -725,13 +725,13 @@ def __call__(self, image, resize_method="bilinear", resize_antialias=False, inpu
  return images
 
 
-def add_pre_post_process(model, rescale_mode="tf", input_shape=None, post_process=None, featrues=None):
+def add_pre_post_process(model, rescale_mode="tf", input_shape=None, post_process=None, features=None):
  from keras_cv_attention_models.imagenet.eval_func import decode_predictions
 
  input_shape = model.input_shape[1:] if input_shape is None else input_shape
  model.preprocess_input = PreprocessInput(input_shape, rescale_mode=rescale_mode)
  model.decode_predictions = decode_predictions if post_process is None else post_process
  model.rescale_mode = rescale_mode
 
- if featrues is not None:
+ if features is not None:
  model.extract_features = lambda: features