Added eval()

Spijkervet · Sep 22, 2020 · 654f05f · 654f05f
1 parent cd28fd8
commit 654f05f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/testing/logistic_regression.py b/testing/logistic_regression.py
@@ -172,6 +172,7 @@ def test(args, loader, simclr_model, model, criterion, optimizer):
     )
     simclr_model.load_state_dict(torch.load(model_fp, map_location=args.device.type))
     simclr_model = simclr_model.to(args.device)
+    simclr_model.eval()
 
 
     ## Logistic Regression