Closed prashant45 closed 5 years ago
I run a training script on following environment:
tensorflow-gpu==1.12.0
keras==2.2.4
CUDA==9.0
and OOM error occured but no error about topological sort.
The model's graph does not contain any loop so error about topological sort should not occur. Reducing batch size or disabling gpu might suppress the error.
When running the training, before the initial epoch I get a log like:
Although, I don't have any problem with the final model accuracy. I think, this is related to the fact that there are multiple heads in the
Functional
definition of the model. Do you also get the same when you run the script? WhichKeras/tensorflow
version do you use?The versions, I am using are:
tensorflow-gpu: 1.12.0
Keras: 2.2.4
CUDA Version: 9.0.176