Minor fix

SHI-Labs · Jan 15, 2023 · 5b21843 · 5b21843
1 parent a18e760
commit 5b21843
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/evaluate.py b/examples/evaluate.py
@@ -97,7 +97,7 @@ def main():
     num_classes = DATASETS[args.dataset]['num_classes']
     img_mean, img_std = DATASETS[args.dataset]['mean'], DATASETS[args.dataset]['std']
 
-    model = models.__dict__[args.model](img_size=img_size,
+    model = models.__dict__[args.model]('', False, False, img_size=img_size,
                                         num_classes=num_classes,
                                         positional_embedding=args.positional_embedding,
                                         n_conv_layers=args.conv_layers,

diff --git a/examples/main.py b/examples/main.py
@@ -116,7 +116,7 @@ def main():
     num_classes = DATASETS[args.dataset]['num_classes']
     img_mean, img_std = DATASETS[args.dataset]['mean'], DATASETS[args.dataset]['std']
 
-    model = models.__dict__[args.model](img_size=img_size,
+    model = models.__dict__[args.model]('', False, False, img_size=img_size,
                                         num_classes=num_classes,
                                         positional_embedding=args.positional_embedding,
                                         n_conv_layers=args.conv_layers,