Some parts of the code have: `if os.environ["KERAS_BACKEND"] == "torch"`, which should be avoided.
Some parts of the code have:
if os.environ["KERAS_BACKEND"] == "torch", which should be avoided.