changed data format in Conv2D

tdeboissiere / DeepLearningImplementations

Implementation of recent Deep Learning papers

MIT License

1.81k stars 650 forks source link

changed data format in Conv2D #51

Closed shirsenh closed 6 years ago

shirsenh commented 6 years ago

There was a reshape error due to the data format which is default to 'channels_last' in Keras. Changed it to 'channels_first'

tdeboissiere commented 6 years ago

Good catch, thanks. I tested it only with theano, (see README : keras with theano backend). I'll update the README so that it is clearer. I'll close your PR for now and will update once I get the time to test each backend carefully.