tumourkit.segmentation.hovernet.train.TrainManager

class tumourkit.segmentation.hovernet.train.TrainManager(shape: str, log_dir: str, train_dir: str, valid_dir: str, pretrained_path: str, use_cpu: bool, num_classes: int | None = 2)

Bases: Config

Either used to view the dataset or to initialise the main training loop.

Methods

run
run_once	Simply run the defined run_step of the related method once.
view_dataset

run_once(opt, run_engine_opt, log_dir, prev_log_dir=None, fold_idx=0)

Simply run the defined run_step of the related method once.