src/transformers/optimization_tf.py/AdamWeightDecay/get_config
class AdamWeightDecay:    def get_config(self):
        config = super().get_config()
        config.update({"weight_decay_rate": self.weight_decay_rate})
        return config