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

Generalization for resneXt #4

Open
n01z3 opened this issue Dec 11, 2018 · 1 comment
Open

Generalization for resneXt #4

n01z3 opened this issue Dec 11, 2018 · 1 comment

Comments

@n01z3
Copy link

n01z3 commented Dec 11, 2018

Hello!
Greaty results! How to generalization this layer for resneXt network with groups != 1 in conv layers inside Bottleneck block?

@liuguilin1225
Copy link
Contributor

If you are just using partial conv for padding, setting groups = 1 in line https://github.com/NVIDIA/partialconv/blob/master/models/partialconv2d.py#L59 will work even the main convolution's groups is not 1. As what we want is just a reweighting map and the mask channel size is 1.
However, if your mask channel size is not 1, you can set the groups to be the main convolution's groups value accordingly, and divide the channel size in https://github.com/NVIDIA/partialconv/blob/master/models/partialconv2d.py#L33 by groups as well.

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

2 participants