def get_loaders(self, stage: str):
        """Insert into loaders collate_fn.

        Args:
            stage (str): sage name

        Returns:
            ordered dict with torch.utils.data.DataLoader
        """
        loaders = super().get_loaders(stage)
        for item in loaders.values():
            if hasattr(item.dataset, "collate_fn"):
                item.collate_fn = item.dataset.collate_fn
        return loaders
Insert into loaders collate_fn.

        Args:
            stage (str): sage name

        Returns:
            ordered dict with torch.utils.data.DataLoader
        
examples/detection/custom_runner.py