Merge pull request #186 from OFA-Sys/features/ofa_cn

orig_patch_image_size bug fixed
OFA-Sys · Jul 29, 2022 · a26c835 · a26c835
2 parents b8c93fd + f4beec5
commit a26c835
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/models/ofa/ofa.py b/models/ofa/ofa.py
@@ -385,6 +385,8 @@ def ofa_large_architecture(args):
     args.scale_heads = getattr(args, "scale_heads", False)
     args.scale_resids = getattr(args, "scale_resids", False)
 
+    args.orig_patch_image_size = getattr(args, "orig_patch_image_size", 256)
+
 
 @register_model_architecture("ofa", "ofa_base")
 def ofa_base_architecture(args):