tlt.models.image_classification.image_classification_model.ImageClassificationModel.train

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

Train the model using the specified dataset