def from_config(cls, config):
        """Creates an optimizer from its config with WarmUp custom object."""
        custom_objects = {"WarmUp": WarmUp}
        return super(AdamWeightDecay, cls).from_config(config, custom_objects=custom_objects)
Creates an optimizer from its config with WarmUp custom object.
dpkd/transformers/src/transformers/optimization_tf.py