""" Copyright (c) 2024-present Naver Cloud Corp. This source code is licensed under the license found in the LICENSE file in the root directory of this source tree. """ import torch def print_once(message): if not torch.distributed.is_initialized() or torch.distributed.get_rank() == 0: print(message) def pretty(d, indent=0): for key, value in d.items(): print_once("\t" * indent + str(key)) if isinstance(value, dict): pretty(value, indent + 1) else: print_once("\t" * (indent + 1) + str(value))