orig_patch_image_size bug fixed

OFA-Sys · Jul 29, 2022 · f4beec5 · f4beec5
1 parent b8c93fd
commit f4beec5
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):