提交 266db4d8 编写于 作者: H HydrogenSulfate

fix classification bug

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