item()返回loss的值,叠加之后算出总loss,最后再除以mini-batches的数量,取loss平均值。
running_loss += loss.item(),pytorch中的item()
最新推荐文章于 2024-03-01 17:45:00 发布
item()返回loss的值,叠加之后算出总loss,最后再除以mini-batches的数量,取loss平均值。