### Work with ptl
- In ```ControlNet/ldm/models/diffusion/ddpm.py``` Line 591 ```def on_train_batch_start(self, batch, batch_idx): ```
- In ```ControlNet/cldm/logger.py``` Line 74 ```def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, dataloader_idx): ```