def _save_model_args(self, output_dir):
    os.makedirs(output_dir, exist_ok=True)
    self.args.save(output_dir)
def _save_model_args(self, output_dir):
    os.makedirs(output_dir, exist_ok=True)
    self.args.save(output_dir)