Method on_train_batch_end

cldm/logger.py:87–90

(self, trainer, pl_module, outputs, batch, batch_idx)

    # def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, dataloader_idx):
    def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx):
        if not self.disabled:
            # if pl_module.current_epoch % 20 == 0:
            self.log_img(pl_module, batch, batch_idx, split="train")