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

Tensorflow plans? #8

Closed danbri closed 7 years ago

danbri commented 7 years ago

What are the prospects of this running (someday) with TF backend? Any plans in that direction?

noting

https://github.com/coxlab/prednet/pull/5

https://github.com/coxlab/prednet/issues?utf8=✓&q=is%3Aissue%20tensorflow

moliusimon commented 7 years ago

Hello danbri, not the developer but have been testing this method on tensorflow (had an incompatible version of theano) and seems to work out of the box.

For it to work, configure keras to use "image_dim_ordering" = "th" and "backend" = "tensorflow".

bill-lotter commented 7 years ago

Hi danbri and moliusimon, sorry about the delay, but it should work for both backends and both image_dim_orderings now.