Method training_step

sparsh/tactile_ssl/algorithm/module.py:13–14

(self, batch: Dict[str, Any], batch_idx: int)

class Module(ABC):
    @abstractmethod
    def training_step(self, batch: Dict[str, Any], batch_idx: int) -> Dict:
        raise NotImplementedError

    @abstractmethod
    def validation_step(self, batch: Dict[str, Any], batch_idx: int) -> Dict: