Exact versions of dependencies for pix2pix net?

tdeboissiere / DeepLearningImplementations

Implementation of recent Deep Learning papers

MIT License

1.81k stars 650 forks source link

Exact versions of dependencies for pix2pix net? #42

Closed AlexanderFabisch closed 7 years ago

AlexanderFabisch commented 7 years ago

Hi,

I am trying to train the pix2pix model. Unfortunately I get errors with several combinations of Theano / Keras and Tensorflow / Keras. Which versions and which dimension order should I use? Is it documented somewhere?

tdeboissiere commented 7 years ago

I have not documented it sorry. Should run with theano 0.9 or tensorflow 1.0 and keras 1.2.2

AlexanderFabisch commented 7 years ago

OK, does not work for me with tensorflow because I use CUDA 7.5 but it seems to work with theano 0.9. At least I did not get an error yet. Should I open a pull request to update the README or can you do that?