torch_mist.utils.evaluation.model

Module Contents

Functions

evaluate(, batch_size, num_workers, Dict[str, float]])

torch_mist.utils.evaluation.model.evaluate(model: torch.nn.Module, method: str, data: torch_mist.utils.data.utils.TensorDictLike, device: torch.device = torch.device('cpu'), batch_size: int | None = None, num_workers: int = 0) → float | Dict[str, float]