ts2.models.pretrained#
Eval trainers that predict activity from a pretrained encoder.
TS2 trainer for MtM, prompting the forward with the task's mask-mode token. |
|
TS2 trainer for NDT Stitch. |
Eval trainers that predict activity from a pretrained encoder.
TS2 trainer for MtM, prompting the forward with the task's mask-mode token. |
|
TS2 trainer for NDT Stitch. |