torchrecipes/vision/image_generation/module/gan.py [78:85]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def training_step(
        self,
        batch: Batch,
        batch_idx: int,
        optimizer_idx: int,
        *args: Any,
        **kwargs: Any,
    ) -> TrainOutput:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



torchrecipes/vision/image_generation/module/infogan.py [133:140]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def training_step(
        self,
        batch: Batch,
        batch_idx: int,
        optimizer_idx: int,
        *args: Any,
        **kwargs: Any,
    ) -> TrainOutput:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -