examples/tensorflow/language-modeling/run_mlm.py/SavePretrainedCallback/__init__
class SavePretrainedCallback:    def __init__(self, output_dir, **kwargs):
        super().__init__()
        self.output_dir = output_dir