40 lines
1.1 KiB
Python
40 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")
|