提交 e7c24e04 authored 作者: Glenn Jocher's avatar Glenn Jocher

half() fix #66

上级 11199495
...@@ -45,6 +45,7 @@ def test(data, ...@@ -45,6 +45,7 @@ def test(data,
training = False training = False
else: # called by train.py else: # called by train.py
device = next(model.parameters()).device # get model device device = next(model.parameters()).device # get model device
half = False
training = True training = True
# Configure # Configure
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论