Function get_output_dir

utils.py:155–188

Initialize the output dir.

(args: argparse)

def get_output_dir(args: argparse):
    """
    Initialize the output dir.
    """

    if args.checkpoint_dir is None:
        checkpoint_dir = os.path.join(args.root_path, 'checkpoints', args.dataset.lower(), args.backbone.lower(), args.method.lower())

        name_str = f'-n_way={args.train_way}' \
                   f'-n_shot={args.num_shot}' \
                   f'-lr={args.lr}' \
                   f'-scheduler={args.scheduler}' \
                   f'-dropout={args.dropout}'

        checkpoint_dir = os.path.join(checkpoint_dir, name_str)
    else:
        checkpoint_dir = args.checkpoint_dir

    if args.eval:
        return checkpoint_dir

    while os.path.exists(checkpoint_dir):
        checkpoint_dir += f'-{np.random.randint(100)}'

    os.makedirs(checkpoint_dir, exist_ok=True)

    # write args to a file
    with open(os.path.join(checkpoint_dir, "args.txt"), 'w') as f:
        for key, value in vars(args).items():
            f.write('%s:%s\n' % (key, value))

    print("=> Checkpoints will be saved at:\n", checkpoint_dir)

    return checkpoint_dir


def load_weights(model: torch.nn.Module, pretrained_path: str):