glenn-jocher commited on
Commit
e7c24e0
·
1 Parent(s): 1119949

half() fix #66

Browse files
Files changed (1) hide show
  1. test.py +1 -0
test.py CHANGED
@@ -45,6 +45,7 @@ def test(data,
45
  training = False
46
  else: # called by train.py
47
  device = next(model.parameters()).device # get model device
 
48
  training = True
49
 
50
  # Configure
 
45
  training = False
46
  else: # called by train.py
47
  device = next(model.parameters()).device # get model device
48
+ half = False
49
  training = True
50
 
51
  # Configure