autodl-projects/xautodl/config_utils/share_args.py

import os, sys, time, random, argparse


def add_shared_args(parser):
    # Data Generation
    parser.add_argument("--dataset", type=str, help="The dataset name.")
    parser.add_argument("--data_path", type=str, help="The dataset name.")
    parser.add_argument(
        "--cutout_length", type=int, help="The cutout length, negative means not use."
    )
    # Printing
    parser.add_argument(
        "--print_freq", type=int, default=100, help="print frequency (default: 200)"
    )
    parser.add_argument(
        "--print_freq_eval",
        type=int,
        default=100,
        help="print frequency (default: 200)",
    )
    # Checkpoints
    parser.add_argument(
        "--eval_frequency",
        type=int,
        default=1,
        help="evaluation frequency (default: 200)",
    )
    parser.add_argument(
        "--save_dir", type=str, help="Folder to save checkpoints and log."
    )
    # Acceleration
    parser.add_argument(
        "--workers",
        type=int,
        default=8,
        help="number of data loading workers (default: 8)",
    )
    # Random Seed
    parser.add_argument("--rand_seed", type=int, default=-1, help="manual seed")