forked from miracleyoo/pytorch-lightning-template
-
Notifications
You must be signed in to change notification settings - Fork 0
/
opt.py
23 lines (20 loc) · 1.32 KB
/
opt.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import argparse
import sys
def get_args():
parser = argparse.ArgumentParser()
parser.add_argument('--cfg_file', type=str, default=None, help='specify the config for training')
parser.add_argument("--mode", default="train", choices=["train", "reconstruction"])
parser.add_argument("--group", default="") # 用于区分不同的实验组
parser.add_argument("--name", default="") # 用于区分不同的实验
parser.add_argument("--log_dir", default='logs', help="path to log into")
parser.add_argument("--ckpt_path", default=None, help="path to checkpoint to restore")
parser.add_argument("--device_ids", default="0", type=lambda x: list(map(int, x.split(','))),
help="Names of the devices comma separated.")
parser.add_argument("--seed", default=3407, type=int, help="random seed")
parser.add_argument("--is_logger_enabled", default=True, type=bool,
help="whether to enable logger")
parser.add_argument("opts", help="Modify config options using the command-line", default=None,
nargs=argparse.REMAINDER)
# 添加半精度
parser.add_argument("--precision", default=32, type=int, help="precision")
return parser.parse_args()