Browse Source

half() fix #66

5.0
Glenn Jocher 4 years ago
parent
commit
e7c24e04ac
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      test.py

+ 1
- 0
test.py View File

@@ -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

Loading…
Cancel
Save