e81f54bb33f4a41c2ebdfdda41e5d70b52c32c87,pyannote/audio/embedding/base_autograd.py,SequenceEmbeddingAutograd,restart,#Any#Any#Any#,188

Before Change

        // optimizer (configuration)
        optimizer_pkl = _.OPTIMIZER_PKL.format(log_dir=log_dir, epoch=epoch)
        with open(optimizer_pkl, mode="rb") as fp:
            state = pickle.load(fp)
        optimizer = keras.optimizers.deserialize(
            state["optimizer_config"],
            custom_objects=CUSTOM_OBJECTS)

        // optimizer (weights)
        embedding.compile(optimizer=optimizer,
                          loss=cls._gradient_loss)

After Change

                                                       epoch=epoch)

        // TODO update this code once keras > 2.0.4 is released
        try:
            embedding = keras.models.load_model(
                weights_h5, custom_objects=CUSTOM_OBJECTS,
                compile=True)
        except TypeError as e:
            embedding = keras.models.load_model(
                weights_h5, custom_objects=CUSTOM_OBJECTS)

        embedding.epoch = epoch

        return embedding

Instances