PyTorch、Caffe绘制训练过程的accuracy和loss曲线
Caffe
PyTorch
#定义两个数组 Loss_list = [] Accuracy_list = [] Loss_list.append(train_loss / (len(train_dataset))) Accuracy_list.append(100 * train_acc / (len(train_dataset))) #我这里迭代了200次，所以x的取值范围为(0，200)，然后再将每次相对应的准确率以及损失率附在x上 x1 = range(0, 200) x2 = range(0, 200) y1 = Accuracy_list y2 = Loss_list plt.subplot(2, 1, 1) plt.plot(x1, y1, 'o-') plt.title('Test accuracy vs. epoches') plt.ylabel('Test accuracy') plt.subplot(2, 1, 2) plt.plot(x2, y2, '.-') plt.xlabel('Test loss vs. epoches') plt.ylabel('Test loss') plt.show() plt.savefig("accuracy_loss.jpg")希望我的分享可以对大家后期的学习起到帮助，可以及时的调整参数，得到更好的模型。
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请联系我们举报，一经查实，本站将立刻删除。
发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/228032.html原文链接：https://javaforall.net
