def log_image(self, key: str, global_step: int, epoch: int, img_tensor: Tensor):
    raise NotImplementedError
def log_image(self, key: str, global_step: int, epoch: int, img_tensor: Tensor):
    raise NotImplementedError