POYOPlusCalibrateTrainer#

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

Bases: pretrain.models.poyo_plus.poyo_plus_multitask_pretrain.POYOPlusMultitaskPretrain

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.