提交 1c91e17c 编写于 作者: D dongshuilong 提交者: gaotingquan

fix dali distributed eval bug

上级 66227977
...@@ -91,7 +91,7 @@ def classification_eval(engine, epoch_id=0): ...@@ -91,7 +91,7 @@ def classification_eval(engine, epoch_id=0):
paddle.distributed.all_gather(pred_list, out) paddle.distributed.all_gather(pred_list, out)
pred = paddle.concat(pred_list, 0) pred = paddle.concat(pred_list, 0)
if accum_samples > total_samples: if accum_samples > total_samples and not engine.use_dali:
pred = pred[:total_samples + current_samples - pred = pred[:total_samples + current_samples -
accum_samples] accum_samples]
labels = labels[:total_samples + current_samples - labels = labels[:total_samples + current_samples -
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册