提交 9c4f9205 编写于 作者: D dengkaipeng

fit for yolov3

上级 ca7c0a9a
......@@ -470,7 +470,7 @@ def run(model, loader, mode='train'):
start = time.time()
for idx, batch in enumerate(loader()):
outputs, losses = getattr(model, mode)(
losses, _ = getattr(model, mode)(
batch[0], batch[1], device='gpu', device_ids=device_ids)
total_loss += np.sum(losses)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册