EVOLUTION-MANAGER

Edit File: __init__.py

import os from .common import * _BACKEND = "tensorflow" if "KERAS_BACKEND" in os.environ: _backend = os.environ["KERAS_BACKEND"] backends = { "cntk", "tensorflow", } assert _backend in backends _BACKEND = _backend if _BACKEND == "cntk": from .cntk_backend import * elif _BACKEND == "tensorflow": from .tensorflow_backend import * else: raise ValueError("Unknown backend: " + str(_BACKEND))