ertk.pytorch.classification.train_mtl_model

ertk.pytorch.classification.train_mtl_model(model: MTLModel, tasks: dict[int, int], data: MTLDataModule | None, train_data: DataLoader | None = None, valid_data: DataLoader | None = None, epochs: int = 200, device: str = 'cuda:0', verbose: int = 0)