cellflow.training.CellFlowTrainer.train

CellFlowTrainer.train(dataloader, num_iterations, valid_freq, valid_loaders=None, monitor_metrics=[], callbacks=[])

Trains the model.

Parameters:

• dataloader (TrainSampler | OOCTrainSampler) – Dataloader used.

• num_iterations (int) – Number of iterations to train the model.

• valid_freq (int) – Frequency of validation.

• valid_loaders (dict[str, ValidationSampler] | None) – Valid loaders.

• callbacks (Sequence[BaseCallback]) – Callback functions.

• monitor_metrics (Sequence[str]) – Metrics to monitor.

Return type:

OTFlowMatching | GENOT

Returns:

: The trained model.