search

tscohen / GrouPy

Group Equivariant Convolutional Neural Networks
http://ta.co.nl
Other
349 stars 85 forks source link

PyTorch implementation #7

Open adambielski opened 6 years ago

adambielski commented 6 years ago

I implemented group equivariant convolutions in PyTorch. Code includes tests for filter transformation and equivariance. I also tested this implementation by training on CIFAR-10 with ResNet-18 architecture https://github.com/adambielski/pytorch-gconv-experiments

tscohen commented 6 years ago

Hi Adam,

Thank you for your contribution. It is much appreciated.

I am currently in a conversation with my new employer to see how I can continue to contribute to this and other open source projects, so it could take a while to merge.