half() fix #66

5 years ago · e7c24e04ac
parent 111994956e
commit e7c24e04ac
1 changed files with 1 additions and 0 deletions
--- 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