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

fit for yolov3

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