Commit
·
e7c24e0
1
Parent(s):
1119949
half() fix #66
Browse files
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
|