Method configure_optimizers

sparsh/tactile_ssl/algorithm/module.py:21–28

(
        self, num_iterations_per_epoch: int, num_epochs: int
    )

    @abstractmethod
    def configure_optimizers(
        self, num_iterations_per_epoch: int, num_epochs: int
    ) -> Tuple[
        torch.optim.Optimizer,
        Optional[Dict],
        Optional[Dict],
    ]:
        raise NotImplementedError

    def on_train_epoch_end(self, trainer_instance=None):
        pass