fishy.models.deep.ttt_mixin¶

Classes

class fishy.models.deep.ttt_mixin.TTTMixin[source]¶

Bases: object

Mixin to add Test-Time Training (entropy minimization) capability to any PyTorch model.

forward_ttt(x: Tensor, lr: float = 0.001, steps: int = 1) → Tensor[source]¶: Test-Time Training (TTT) via Entropy Minimization. Adapts the model to the specific sample x at inference time.