loss_all = {'train':[],'val':[]}
loss_all['train'].append(loss_trn/len(trainloader))
loss_all['val'].append(loss_val/len(valiateloader))
plt.plot(loss_all['train'])
plt.plot(loss_all['val'])
绘制loss图
最新推荐文章于 2024-07-24 21:01:17 发布