From e7c24e04ac0c792cee9028ac4715b77ac4c9b273 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 15 Jun 2020 00:43:57 -0700 Subject: [PATCH] half() fix #66 --- test.py | 1 + 1 file changed, 1 insertion(+) 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