search

songdejia / Siamese-RPN-pytorch

This is a re-implementation of Siamese-RPN with pytorch, which is CVPR2018 spotlight.
MIT License
379 stars 87 forks source link

Unable to save training model #16

Open donggc opened 5 years ago

donggc commented 5 years ago

Use the following command to train the model, but model is not produced after the training. python train_siamrpn.py --dataroot=/home//projects/firefox-downloads/Siamese-RPN-pytorch-master/vot2013 --lr=0.001 --checkpoint_path=/home//projects/firefox-downloads/Siamese-RPN-pytorch-master/pretrainedModel/weights-0690000.pth.tar

donggc commented 5 years ago

Looking forward to your reply

qchenclaire commented 5 years ago

Use the following command to train the model, but model is not produced after the training. python train_siamrpn.py --dataroot=/home//projects/firefox-downloads/Siamese-RPN-pytorch-master/vot2013 --lr=0.001 --checkpoint_path=/home//projects/firefox-downloads/Siamese-RPN-pytorch-master/pretrainedModel/weights-0690000.pth.tar

Moving https://github.com/songdejia/Siamese-RPN-pytorch/blob/master/code_v1.0/train_siamrpn.py#L247 to the loop of https://github.com/songdejia/Siamese-RPN-pytorch/blob/master/code_v1.0/train_siamrpn.py#L89 works for me

XiongZhongxia commented 5 years ago

Putting line247-254 in the loop of each epoch will solve this.