69813de15a040584756c5dbfa61468f20c97f8af,lib/model/normalization/normalization_common.py,,,#,488

Before Change

// Update normalization into Keras custom objects
for name, obj in inspect.getmembers(sys.modules[__name__]):
    if inspect.isclass(obj) and obj.__module__ == __name__:
        get_custom_objects().update({name: obj})

After Change

from lib.utils import get_backend


if get_backend() == "amd":
    from keras.backend import normalize_data_format  // pylint:disable=ungrouped-imports
else:
    from tensorflow.python.keras.utils.conv_utils import normalize_data_format

Instances