search

cypw / PyTorch-MFNet

MIT License
253 stars 56 forks source link

Fine-tune with pre-trained model #33

Open Serendipity-LC opened 4 years ago

Serendipity-LC commented 4 years ago

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

liuhui0401 commented 2 years ago

You can use torch.save()