allow functional depthwise conv

rwth-i6 / pytorch-to-returnn

Make PyTorch code runnable within RETURNN

3 stars 6 forks source link

allow functional depthwise conv #128

Closed vieting closed 1 year ago

vieting commented 1 year ago

Allow using groups for _FunctionalConvNd. I also added a test for depthwise convolution with stride 2. This revealed some problems on RETURNN side. It should work with https://github.com/rwth-i6/returnn/pull/1227.

vieting commented 1 year ago

The failing test cases are those already discussed in #125. The new one I added (and all others dealing with convolution) pass, so this PR is ready for review.