made input shapes (None, None, 3)

sayakpaul · Oct 23, 2022 · 6364fd2 · 6364fd2
1 parent 8990a26
commit 6364fd2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/convert_to_tf.py b/convert_to_tf.py
@@ -196,13 +196,13 @@ def main(args):
     dataset_name = args.ckpt_path.split("/")[-2].lower()
     tf_params_path = f"{variant}_{task.lower()}_{dataset_name}.h5"
 
-    tf_model.save_weights(tf_params_path)
-    print(f"Model params serialized to {tf_params_path}.")
+    # tf_model.save_weights(tf_params_path)
+    # print(f"Model params serialized to {tf_params_path}.")
     saved_model_path = tf_params_path.replace(".h5", "")
     tf_model.save(saved_model_path)
     print(f"SavedModel serialized to {saved_model_path}.")
-    push_to_hub_keras(tf_model, repo_path_or_name=f"sayakpaul/{saved_model_path}")
-    print("Model pushed to Hugging Face Hub.")
+    # push_to_hub_keras(tf_model, repo_path_or_name=f"sayakpaul/{saved_model_path}")
+    # print("Model pushed to Hugging Face Hub.")
 
 
 def parse_args():

diff --git a/create_maxim_model.py b/create_maxim_model.py
@@ -4,7 +4,7 @@
 from maxim.configs import MAXIM_CONFIGS
 
 
-def Model(variant=None, input_resolution=(256, 256), **kw) -> keras.Model:
+def Model(variant=None, input_resolution=(None, None), **kw) -> keras.Model:
     """Factory function to easily create a Model variant like "S".
 
     Args: