search

QizeYang / PAUL

92 stars 16 forks source link

pre-train model #6

Open ngocson1804 opened 5 years ago

ngocson1804 commented 5 years ago

what data set did you train the model on the driver? it is MSMT17, is this right?. I want to use this model but there is an error '../../imagenet_models/resnet50-19c8e357.pth'. thank you!

QizeYang commented 5 years ago

Hi! You are right and that error was caused by the default value of the parameters of PatchNetUn. If possible, you can change pretrained=True to pretrained=False.
https://github.com/QizeYang/PAUL/blob/3aacd5908b83b4904b325d218f7f3f5a3bfa7699/src/models/patchnet.py#L140

Thank you.