def log_miou(miou, classes):
logger = logging.get_logger()
padding = max(len(cls) for cls in classes)
for miou_i, class_i in zip(miou, classes):
logger.info(("{:>" + str(padding) + "} : {:.3f}").format(
class_i, miou_i.item()))
def log_info(msg, *args, **kwargs):
if distributed.get_rank() == 0:
logging.get_logger().info(msg, *args, **kwargs)
def log_debug(msg, *args, **kwargs):
if distributed.get_rank() == 0:
logging.get_logger().debug(msg, *args, **kwargs)
