tlt.models.text_generation.text_generation_model.TextGenerationModel.train

abstract TextGenerationModel.train(dataset: BaseDataset, output_dir, epochs=1, initial_checkpoints=None, do_eval=True)

Train the model using the specified dataset