fix: vitclip16 requires 224 input size

jolibrain · Mar 7, 2023 · 8b8bb67 · 8b8bb67
1 parent d83f040
commit 8b8bb67
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/options/base_options.py b/options/base_options.py
@@ -898,6 +898,14 @@ def _after_parse(self, opt, set_device=True):
                 "Bounding box class selection requires --data_sanitize_paths"
             )
 
+        # vitclip16 projector only works with input size 224
+        if opt.D_proj_network_type == "vitclip16":
+            if opt.D_proj_interp != 224:
+                warnings.warn(
+                    "ViT-B/16 (vitclip16) projector only works with input size 224, setting D_proj_interp to 224"
+                )
+            opt.D_proj_interp = 224
+
         self.opt = opt
 
         return self.opt