提交 0b514dac authored 作者: Glenn Jocher's avatar Glenn Jocher

FP16 test loss bug fix

上级 a1748a8d
......@@ -104,7 +104,7 @@ def test(data,
# Compute loss
if training: # if model has loss hyperparameters
loss += compute_loss(train_out, targets, model)[1][:3] # GIoU, obj, cls
loss += compute_loss([x.float() for x in train_out], targets, model)[1][:3] # GIoU, obj, cls
# Run NMS
t = torch_utils.time_synchronized()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论