use tf1.compat mode for keras

This commit is contained in:
cneud 2024-03-19 22:20:19 +01:00
parent 5fcfe61afc
commit 9d24150031

View file

@ -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