POSSMCalibrateTrainer#

class ts3.models.transductive.POSSMCalibrateTrainer(cfg, rank, world_size)[source]#

Bases: pretrain.models.possm.possm_multitask_pretrain.POSSMMultitaskPretrain

setup(ckpt)[source]#

The parent’s setup, plus the finetuning strategy the eval trainers apply.

cfg.finetuning is read by ts1/ts2’s eval trainers but not by the pretrain trainer, so calibration inheriting the pretrain one would accept the block and ignore it.

train_epoch()[source]#

Perform a training epoch.