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

Python 3 / Tensorflow 1.12.0 Compatibility #49

Closed berkgercek closed 5 years ago

berkgercek commented 5 years ago

This is a set of changes to the base code of PredNet which allows it to run successfully on python 3.6, using the latest Tensorflow (TF 1.12.0) and Keras (2.2.4) installations available. The major changes to make prednet work in python 3 are:

EDIT// As a note this has been tested on CUDA 9.0 (which I believe is the latest version supported by tensorflow) and CUDNN 7.4.1.5

Hope this will be as useful to others as it will be to me!

bill-lotter commented 5 years ago

Thanks so much for this and sorry it took me awhile to test!