image size option

2023-10-15 23:48:14 +01:00
parent cf9c34c801
commit 62746e7f03
3 changed files with 6 additions and 3 deletions
@@ -239,6 +239,7 @@ def main(args):
            label_smoothing=args.smoothing, num_classes=args.nb_classes)
    
    model = models_vit.__dict__[args.model](
+        img_size=args.input_size,
        num_classes=args.nb_classes,
        drop_path_rate=args.drop_path,
        global_pool=args.global_pool,