(loss,epochs)
| 39 | |
| 40 | |
| 41 | def draw(loss,epochs): |
| 42 | plt.plot(range(1,epochs + 1),loss,label = 'trainLoss') |
| 43 | plt.legend() |
| 44 | plt.title('AE-LOSS') |
| 45 | plt.savefig('logs/figure.png') |
| 46 | # plt.show() |
| 47 | |
| 48 | |
| 49 | def save_checkPoint(model,optimizer,filename): |
nothing calls this directly
no outgoing calls
no test coverage detected