:param gen_loss: :param disc_loss: :return:
(loss)
| 52 | pass |
| 53 | |
| 54 | def draw(loss): |
| 55 | """ |
| 56 | :param gen_loss: |
| 57 | :param disc_loss: |
| 58 | :return: |
| 59 | """ |
| 60 | plt.plot(range(1, len(loss) + 1), loss, label='VAELoss') |
| 61 | plt.legend() |
| 62 | plt.title('VAE-LOSS') |
| 63 | plt.savefig('logs/figure.png') |
| 64 | |
| 65 | def save_images(imgs,epoch,step): |
| 66 | imgs = imgs.detach().cpu().numpy() |
nothing calls this directly
no outgoing calls
no test coverage detected