我的模型的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')