-
Notifications
You must be signed in to change notification settings - Fork 232
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
How to fine tune pretrained pascal person model on your own data? #33
Comments
Well, I think both --model-restore and --schp-restore are used for resume training, not well fit for finetune, you can use the default value, and then, add: |
@Julymycin Thanks for your suggestion! I tried that, but I am getting this error : size mismatch for module.decoder.conv4.weight: copying a param with shape torch.Size([7, 256, 1, 1]) from checkpoint, the shape in current model is torch.Size([3, 256, 1, 1]). My data has 3 classes, and the pascal person dataset has 7. That can be the issue, but I'm struggling to figure out how to resolve. it. |
@rkhilnani9 Well, there are two possibly helpful options for you.
|
Dear Sir I followed your step and I can load LIP pretrained model so far, however the errors shows
I checked the model parameter and buffers on gpu using next(model.parameters()).is_cuda() and it return True |
I just solved the problem, the error happend becuase I modified the final fusion layer conv outputs to 3 nodes, and the weight I changed is on cpu, so need to |
Trying to use exp-schp-201908270938-pascal-person-part.pth model for fine tuning on data with number of classes = 3.
Where do I plug in the weights of the model?
Both --model-restore=./pretrain_model/exp-schp-201908270938-pascal-person-part.pth and --schp-restore=./pretrain_model/exp-schp-201908270938-pascal-person-part.pth are throwing errors.
--imagenet-pretrain has to be ./pretrain_model/resnet101-imagenet.pth. So where does the weights of the pascal pretrained model fit in train.py?
The text was updated successfully, but these errors were encountered: