TentEngine#

class torch_ttt.engine.tent_engine.TentEngine(model: Module, optimization_parameters: Dict[str, Any] = {})[source]#

TENT: Fully test-time adaptation by entropy minimization.

Parameters:

  • model (torch.nn.Module) – The model to adapt.

  • optimization_parameters (dict) – Hyperparameters for adaptation.

Reference:

“TENT: Fully Test-Time Adaptation by Entropy Minimization” Dequan Wang, Evan Shelhamer, et al.

ttt_forward(inputs) → Tuple[Tensor, Tensor][source]#

Forward pass and entropy loss computation.