-
Notifications
You must be signed in to change notification settings - Fork 474
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
Stride for layer4 in resnet #6
Comments
You can check the convert_from_tensorflow branch now. |
Thanks, I will try it. |
It's now significantly slower (because the stride 1 needs more computation) than tf-faster-rcnn, I will investigate what's the bottleneck. (0.31 -> 0.37 s per iter) |
@ruotianluo I'm trying to train a resnet101+faster rcnn model starting from image net weights. Changing the stride in layer4 from 2->1 does indeed increase the training time. Did you ever resolve the bottleneck? |
@Feynman27 I haven't profiled the code in detail. But it should be faster after commit df3cc05 |
@philokey, I will create a new branch for this so that the current pretrained won't be messed up.
This issue was originally opened in ruotianluo/pytorch-resnet#2
The text was updated successfully, but these errors were encountered: