-
Notifications
You must be signed in to change notification settings - Fork 4.1k
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
Training time for ZF #192
Comments
If you are using end2end training, have a good SSD and compiled Caffe with latest cuDNN, training should be much faster than 200 ms per iteration. |
@happyharrycn Which version of caffe do you use ? I tried to build the caffe fork of this repo with cuda 7.5 and cuDNN v5 and it didn't work. |
I am using the caffe main branch (cuDNN v5 is supported on May 17th). You will need to merge the RoI pooling and the smoothed l1 loss layers from Ross's caffe version. |
@happyharrycn I have done a PR 4163 that merge the both layers to caffe and I am working on porting the code to python 3.4 py-faster-rcnn fork. Can I have the link to Ross's caffe version please ? |
@Austriker I have commented on your PR 4163. It should compile and work now. |
@happyharrycn Great thank you ! I will test the python 3 port ! It will avoid a lot of maintenance problem having only one caffe ! |
How long can I expect faster_rcnn to take to train ZF on VOC2007? I am getting about 4.5hours on a TitanX, which seems pretty long. (The training rate is 0.242s / iter)
The text was updated successfully, but these errors were encountered: