18 lines
1.1 KiB
Python
18 lines
1.1 KiB
Python
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')
|