Added missing stride option to conv2d

junaidmalik09 / fastonn

FastONN - Python based open-source GPU implementation for Operational Neural Networks

GNU General Public License v3.0

22 stars 9 forks source link

Added missing stride option to conv2d #3

Closed ilia-chelak closed 2 years ago

ilia-chelak commented 2 years ago

Hi! Thanks for your amazing work. I'm currently trying to experiment with it for the metric learning task and found out that the stride option is missing in the conv2d list of options in the forward pass of the SelfONNLayer class. So fixed it here.