LOLCATTrainer

class ts3.models.supervised.LOLCATTrainer(cfg, rank, world_size)

Bases: core.trainer.BaseTrainer

Supervised training of LOLCAT, the one TS3 model fit on the region labels.

setup(ckpt)

Set up the trainer.

This method is called after setting up distributed, loading the checkpoint, and setting up logging.

Parameters:

ckpt (dict | None) – The checkpoint dictionary, if any.

train_epoch()

Perform a training epoch.

val_epoch()

Perform a validation epoch.

Called every val.every_n_epochs epochs, and at the end of training.

Returns:

bool, True if early stopping should be triggered, False (or falsey) otherwise.

Return type:

early_stop

test()

Perform a test epoch.

Called at the end of training if test.enable is True.