Open Nqabz opened 6 years ago
Same issue here!
Tensorflow 1.4.1
Keras 2.1.5
Python 3.5.5
CUDA 8.0
This issue also appears to me when I have two models sharing layers - one for prediction, one for training (with CTC loss and auxiliary inputs). ModelCheckpoint
is saving from the training model.
I tried running all your example on 8 Nvidia V100 however I get this error across all of them:
This is in a docker environment with the following versions: **
** Could this be version compatibility or GPU device compatibility? Thanks for any pointers.