ts2.models.pretrained#

Eval trainers that predict activity from a pretrained encoder.

MtMEvalTrainer

TS2 trainer for MtM, prompting the forward with the task's mask-mode token.

NDTStitchEvalTrainer

TS2 trainer for NDT Stitch.