-
Notifications
You must be signed in to change notification settings - Fork 967
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
Add support to groups in SpatialConvolutionMM #438
Comments
Francisco Massa notifications@github.com kirjoitti 22.10.2015 kello 1.05:
But I am just a newcomer in deep learning so you are free to ignore my wishes. Hannu Töyrylä
|
@fmassa the group feature in Caffe and the need to convert caffe models to torch models is what brought me here. If you can add in this feature, it would be amazing! |
@varunagrawal I think the easiest for now would be to add the groups support in the lua level, as done in PyTorch. I unfortunately won't have time in the near future to do it, as I've moved my stack to PyTorch. |
Alright. I wanted to convert some caffe models to torch and then to PyTorch, but the lack of the groups feature in torch is preventing me from doing that. Any ideas on what can I do alternatively? |
Nevermind, found a solution. |
Is there any interest in adding groups support to
SpationConvolutionMM
?Groups behaviour can be simulated by combining a few modules (like
Concat
+Narrow
+SpatialConvolution
, for example), but is it worth it adding dedicated C/Cuda paths for this option ?The only reason why I would see some interest is to allow some Caffe models to work in vanilla
nn
(like AlexNet).If there is interest in it, I can do it.
The text was updated successfully, but these errors were encountered: