Guidance on loading fine-tuned model

[BradMcDanel]

I was successfully able to finetune this example: https://github.com/salesforce/CodeTF/blob/main/test_trainer/test_codet5_trainer.py

However, it is unclear how to save this fine-tuned checkpoint and/or load the checkpoint into a new inference script. Please advise if I have missed documentation on how to do this.

Thanks for the really nice contribution!

[Luxios22]

same issue here, did you make it work?

[chenpe32cp]

same question