diff --git a/test.py b/test.py
index 2f031ab..ee2b0ab 100644
--- a/test.py
+++ b/test.py
@@ -45,6 +45,7 @@ def test(data,
         training = False
     else:  # called by train.py
         device = next(model.parameters()).device  # get model device
+        half = False
         training = True
 
     # Configure