Save generator and load it only for prediction

[agelosk]

Hello,

Thank you for your implementation of cycle gans, it is very clear. I would like to ask if there is a way to save the generators every 500 iterations (exactly when they predict the test images) so I can load them in a different moment and only perform prediction in a specific test set with the loaded model (in a new code, independent of cycle_gan.py)

Thank you,
Agelos

[vpj]

Hi,

You can save checkpoints with experiment.save_checkpoint . It will save all the models added to the experiment as dictionary with experiment.add_pytorch_models.

When starting a new experiment you can load specified models with experiment.load_models. Note that the model names should match between the experiment used to save models and load models.

Documentation links

• save_checkpoint
• add_pytorch_models
• load_models

Join our slack channel for discussions: https://join.slack.com/t/labforml/shared_invite/zt-egj9zvq9-Dl3hhZqobexgT7aVKnD14g/

[vpj]

18c96c5...98b659f

@agelosk this commit does saving and loading of checkpoints. Hope it helps you