examples/tensorflow/language-modeling/run_clm.py/SavePretrainedCallback/on_epoch_end
class SavePretrainedCallback:    def on_epoch_end(self, epoch, logs=None):
        self.model.save_pretrained(self.output_dir)