RRREvalTrainer#

class ts1.models.pretrained.RRREvalTrainer(cfg, rank, world_size)[source]#

Bases: ts1.ts1_eval_trainer.TS1EvalTrainer

TS1 eval trainer plus the V hand-off.

The load must follow configure_readout, which creates V – hence the super() call first.

link_model(model, ckpt)[source]#

Link datasets to the model and configure the readout head.

Attaches model-specific transforms and input_fn to the train/val/test dataset pipelines, then calls model.link_datasets and model.configure_readout to wire up the readout head for the current task.

Parameters:

model (RRRDecoder) – The instantiated model to link.
ckpt (dict | None) – Checkpoint dictionary, passed through for subclass use.