search

mchong6 / JoJoGAN

Official PyTorch repo for JoJoGAN: One Shot Face Stylization
MIT License
1.42k stars 206 forks source link

save finetuned model #6

Closed AK391 closed 2 years ago

AK391 commented 2 years ago

are the finetuned models saved currently in the colab?

mchong6 commented 2 years ago

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?

AK391 commented 2 years ago

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

mchong6 commented 2 years ago

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

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

AK391 commented 2 years ago

@mchong6 ok thanks