torchrecipes/vision/image_classification/module/image_classification.py [222:227]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self,
        batch: Union[List[torch.Tensor], Mapping[str, torch.Tensor]],
        batch_idx: int,
        *args: Any,
        **kwargs: Any,
    ) -> Dict[str, Any]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



torchrecipes/vision/image_classification/module/image_classification.py [241:246]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        self,
        batch: Union[List[torch.Tensor], Mapping[str, torch.Tensor]],
        batch_idx: int,
        *args: Any,
        **kwargs: Any,
    ) -> Dict[str, Any]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -