search

mkocabas / CoordConv-pytorch

Pytorch implementation of CoordConv introduced in 'An intriguing failing of convolutional neural networks and the CoordConv solution' paper. (https://arxiv.org/pdf/1807.03247.pdf)
397 stars 50 forks source link

Alternative implementation for PyTorch #3

Closed leVirve closed 6 years ago

leVirve commented 6 years ago

This is another implementation for PyTorch with the tensor shape auto-inference from the inputs. I keep it as a reference for you, however, may not need to merge this but you can also re-arrange it.

mkocabas commented 6 years ago

LGTM, thanks. I'll merge it directly. People can decide on which one to use.