Loss curve visualization

[MayuOshima]

Hello, thanks for sharing this interesting work! At lines 360 and 361 in train.py, there exist

                torch.save(generator_losses, os.path.join(opt.output_dir, 'generator.losses'))
                torch.save(discriminator_losses, os.path.join(opt.output_dir, 'discriminator.losses'))

Could you please tell me how to load the saved generator.losses file and discriminator.losses file? I have tried torch.load, pickle.load, but none of them work.

[MayuOshima]

Or is there any other saved file I can use to visualize the loss curve?

[marcoamonteiro]

Hi,

You can use tensorboard to visualize the loss curves. Just run tensorboard and point --logdir to the output_dir. Does that work for you?

[marcoamonteiro]

Closing due to inactivity. Feel free to reopen if you've got further questions.

[jingwang97]

Hi， I try to use tensorboard to visualize the loss curves,but tensorboard shows'No dashboards are active for the current data set.' the loss curve are saved in .losses file?