def log_image(self, key, image_key, step, epoch, img_tensor, **kwargs):
    if (key not in self.loggers):
        self.init_logger(key)
    self.loggers[key].log_image(image_key, step, epoch, img_tensor, **kwargs)
def log_image(self, key, image_key, step, epoch, img_tensor, **kwargs):
    if (key not in self.loggers):
        self.init_logger(key)
    self.loggers[key].log_image(image_key, step, epoch, img_tensor, **kwargs)