trainer

trainer#

class pointrix.engine.default_trainer.DefaultTrainer(cfg: Config, exp_dir: Path, name: str)#

Bases: BaseTrainer

The default trainer class for training and testing the model.

Parameters:

  • cfg (dict) – The configuration dictionary.

  • exp_dir (str) – The experiment directory.

  • device (str, optional) – The device to use, by default “cuda”.

test(model_path=None) → None#

The testing method for the model.

train_loop() → None#

The training loop for the model.

train_step(batch: List[dict]) → None#

The training step for the model.

Parameters:

batch (dict) – The batch data.

Contents