Fine-tune with pre-trained model

[Serendipity-LC]

Hi！How to save the fine-tuned model as a .pth file？

[liuhui0401]

You can use torch.save()