Load trained model weigths

[ozanpkr]

Hi @likyoo ,

I study with yoru repo for my project.I have been added to new features to your repo.I'll share it with you when I'm done.

But I have a significant question;

How can I load weigths after training operation?

[likyoo]

just torch.load. If you want to only save state_dict for your model, you can modify here.

Looking forward to your contribution !