search

coxlab / prednet

Code and models accompanying "Deep Predictive Coding Networks for Video Prediction and Unsupervised Learning"
https://arxiv.org/abs/1605.08104
MIT License
759 stars 259 forks source link

new backend dim ordering #5

Closed EderSantana closed 7 years ago

EderSantana commented 7 years ago

Fix #3 for recent versions of keras

EderSantana commented 7 years ago

recent versions of keras now uses Tensorflow b01c dim ordering as default, this sets the ordering back to theano's, which is what the codebase is adapted for.

LanaSina commented 7 years ago

This can also be solved by modifying the keras.json file in .keras mine had as default

"image_dim_ordering": "tf",

I just changed it back to th

EderSantana commented 7 years ago

yup

bill-lotter commented 7 years ago

Hi Eder, sorry for the long delay in this and thanks a lot for your help. I've made it now so that it can handle both image_dim_orderings and backends.