fix

nocotan · Nov 9, 2018 · 7f3faff · 7f3faff
1 parent 51db975
commit 7f3faff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/infer.py b/infer.py
@@ -39,7 +39,7 @@ def main():
     testloader = DataLoader(testset, batch_size=args.batch_size,
                             shuffle=False, num_workers=2)
 
-    model = config.model.to(device)
+    model = config.get_model().to(device)
 
     weights = glob.glob(os.path.join(args.weights_root, "*.pth.tar"))