INNER CODE UNIT · Python

load_model

horovod/horovod · horovod/_keras/__init__.py:296

def load_model(keras, wrap_optimizer, filepath, custom_optimizers, custom_objects, legacy_opts=False):
    if legacy_opts:
        keras_subclasses = keras.optimizers.legacy.Optimizer.__subclasses__()
    else:
        keras_subclasses = keras.optimizers.Optimizer.__subclasses__()

    horovod_objects = {
        subclass.__name__.lower(): wrap_optimizer(subclass)
        for subclass in keras_subclasses
    }

    if custom_optimizers is not None:
        horovod_objects.update({
            cls.__name__: wrap_optimizer(cls)
            for cls in custom_optimizers
        })

    if custom_objects is not None:

View source record →