Method log_image

utils/logger.py:83–85

(self, tag: str, image: Tensor, step: int)

    @beartype
    def log_image(self, tag: str, image: Tensor, step: int):
        self._tb_print(f"add tb image {tag}: {image.shape}")
        self.writer.add_image(tag, image, step)

    @beartype
    def log_images(