CDVD-TSP How can use the pre-trained model for fine-tuning on a small dataset?

How can use the pre-trained model for fine-tuning on a small dataset?

Open zabir-nabil opened this issue 4 years ago • 1 comments

I suppose, the default main.py training will start training from scratch, how can use one of your pretrained weights and start training with that?

Sep 01 '20 10:09 zabir-nabil

Is it this flag? --pre_train ../pretrain_models/CDVD_TSP_GOPRO.pt

Sep 01 '20 11:09 zabir-nabil