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.

s