def metrics_batch(output, target): pred = output.argmax(dim=1, keepdim=True) corrects=pred.eq(target.view_as(pred)).sum().item() return corrects.
確定！ 回上一頁
