use tf1.compat mode for keras

2 months ago · 9d24150031
parent 5fcfe61afc
commit 9d24150031
1 changed files with 1 additions and 1 deletions
--- a/eynollah/eynollah/train/train.py
+++ b/eynollah/eynollah/train/train.py
@ -1,7 +1,7 @@
 import os
 import sys
 import tensorflow as tf
-from keras.backend.tensorflow_backend import set_session
+from tensorflow.compat.v1.keras.backend import set_session
 import keras, warnings
 from keras.optimizers import *
 from sacred import Experiment