jeremyfix / deeplearning-lectures

Deep learning lecture materials
https://teaching.pages.centralesupelec.fr/deeplearning-lectures-build/
Other
30 stars 13 forks source link

Forgot to reshape in keras tuto #1

Closed jeremyfix closed 5 years ago

jeremyfix commented 5 years ago

In 00-keras.md , when dealing with convolutional layers, we need to reshape the X_train and X_train to (num_train, height, width, 1) and (num_test, height ,width, 1)