save finetuned model

[Ahsen Khaliq]

are the finetuned models saved currently in the colab?

[Min Jin Chong]

The fine-tuned models are not currently saved as they will disappear after the session. There probably is a way to save it directly to your GDrive. Is that something you want?

[Ahsen Khaliq]

@mchong6 oh I actually meant in the colab notebook I could not find the location, let me check again

[Min Jin Chong]

I did not save it in the colab notebook. To save it you can run

torch.save({"g": generator.state_dict()}, f"checkpoint.pt")

[Ahsen Khaliq]

@mchong6 ok thanks