I use Keras-gpu 2.3.1 and tensorflow-gpu 2.0.0. I\'d like to stick to it. I saved my model as .pb.
.pb
My custom loss function:
import tensorflow.