From 5b2184359d1a3cc81c1baf46c89d88059340e9d4 Mon Sep 17 00:00:00 2001 From: Ali Hassani Date: Sun, 15 Jan 2023 16:36:12 -0500 Subject: [PATCH] Minor fix --- examples/evaluate.py | 2 +- examples/main.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/evaluate.py b/examples/evaluate.py index b190a7a..2c0b19c 100644 --- 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 index 1e2558d..3334777 100644 --- 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,