Function write_args_to_tensorboard

codegeex/megatron/initialize.py:305–311

Write arguments to tensorboard.

()

def write_args_to_tensorboard():
    """Write arguments to tensorboard."""
    args = get_args()
    writer = get_tensorboard_writer()
    if writer:
        for arg in vars(args):
            writer.add_text(arg, str(getattr(args, arg)), global_step=args.iteration)


def initialize_wandb_experiment():