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

fix classification bug

上级 5f1c265a
......@@ -98,7 +98,14 @@ def classification_eval(engine, epoch_id=0):
preds = paddle.concat(pred_list, 0)
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 -
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.
先完成此消息的编辑!
想要评论请 注册