tumourkit.classification.train_graphs.train_one_iter

tumourkit.classification.train_graphs.train_one_iter(tr_loader: GraphDataLoader, model: Module, device: str, optimizer: Optimizer, epoch: int, writer: SummaryWriter, num_classes: int, enable_background: bool | None = False) → None

Trains for one iteration, as the name says.