我的模型的epoch是从1开始训练的,画图的时候出现 ValueError: x and y must have same first dimension, but have shapes (2,) and (1,)
ax1.plot(np.arange(epoch + 1), loss[0], '-y', label='ste-model loss')
ax1.plot(np.arange(epoch + 1), loss[1], '-r', label='discriminator loss')
修改为
ax1.plot(np.arange(epoch ), loss[0], '-y', label='ste-model loss')
ax1.plot(np.arange(epoch ), loss[1], '-r', label='discriminator loss')
该博客讲解了如何修正模型训练过程中epoch值绘制图表时的ValueError,通过将np.arange()的参数从(epoch+1)改为(epoch),解决了loss数组在画图时维度不匹配的问题,确保了ste-modelloss和discriminatorloss的正确显示。
1万+

被折叠的 条评论
为什么被折叠?



