fix the extra argument in models.get_model()

tml-epfl · Sep 1, 2021 · 96d1a62 · 96d1a62
1 parent 9786bfa
commit 96d1a62
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eval.py b/eval.py
@@ -48,7 +48,7 @@ def get_args():
 torch.manual_seed(args.seed)
 torch.cuda.manual_seed(args.seed)
 
-model = models.get_model(args.model, n_cls, half_prec, data.shapes_dict[args.dataset], args.n_filters_cnn, args.n_hidden_fc)
+model = models.get_model(args.model, n_cls, half_prec, data.shapes_dict[args.dataset], args.n_filters_cnn)
 model = model.cuda()
 model_dict = torch.load('models/{}.pth'.format(args.model_path))
 if args.early_stopped_model: