Skip to content
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

Closed
ruotianluo opened this issue Sep 8, 2017 · 5 comments
Closed

Stride for layer4 in resnet #6

ruotianluo opened this issue Sep 8, 2017 · 5 comments

Comments

@ruotianluo
Copy link
Owner

ruotianluo commented Sep 8, 2017

@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

@ruotianluo
Copy link
Owner Author

You can check the convert_from_tensorflow branch now.

@philokey
Copy link

philokey commented Sep 9, 2017

Thanks, I will try it.

@ruotianluo
Copy link
Owner Author

ruotianluo commented Sep 9, 2017

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)

@Feynman27
Copy link

Feynman27 commented Oct 3, 2017

@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?

@ruotianluo
Copy link
Owner Author

@Feynman27 I haven't profiled the code in detail. But it should be faster after commit df3cc05

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants