-
Notifications
You must be signed in to change notification settings - Fork 6
/
options.py
103 lines (73 loc) · 3.34 KB
/
options.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
# -*- coding: utf-8 -*-
"""
@Author : zhwzhong
@License : (C) Copyright 2013-2018, hit
@Contact : [email protected]
@Software: PyCharm
@File : options.py
@Time : 2023/2/1 18:25
@Desc :
"""
import os
import yaml
import argparse
from set_config import set_config
from utils import init_distributed_mode, set_random_seed, get_gpu_info
config_parser = parser = argparse.ArgumentParser(description='Config', add_help=False)
parser.add_argument('-c', '--config', default='', type=str, metavar='FILE',
help='YAML config file specifying default arguments')
parser = argparse.ArgumentParser(description='PyTorch PBVS Challenge')
# Hardware
parser.add_argument('--device', default='cuda')
parser.add_argument('--sync_bn', action='store_true')
parser.add_argument('--seed', type=int, default=60)
parser.add_argument('--num_gpus', type=int, default=2)
parser.add_argument('--num_workers', type=int, default=0)
# Optimizer
parser.add_argument('--lr', type=float, default=1e-4)
parser.add_argument('--opt', type=str, default='AdamW')
parser.add_argument('--loss', type=str, default='1*L1')
parser.add_argument('--hdelta', type=float, default=1) # HuberLoss
parser.add_argument('--epochs', type=int, default=120)
parser.add_argument('--sched', default='multistep', type=str) # cosine multistep
parser.add_argument('--weight_decay', type=float, default=0)
parser.add_argument('--warmup_epochs', type=int, default=0)
parser.add_argument('--cooldown_epochs', type=int, default=10)
parser.add_argument('--min_lr', type=float, default=1e-5)
parser.add_argument('--warmup_lr', type=float, default=1e-5) # warmup 初始的LR,warmup-epoch以后变为设定的lr
parser.add_argument('--decay_rate', type=float, default=0.5)
parser.add_argument('--decay_epochs', type=str, default='100')
# Training Stats
parser.add_argument('--resume', action='store_true')
parser.add_argument('--test_only', action='store_true')
parser.add_argument('--start_epoch', type=int, default=0)
parser.add_argument('--checkpoint_hist', type=int, default=10)
parser.add_argument('--load_name', type=str, default='model_best.pth')
parser.add_argument('--scale', type=int, default=4)
parser.add_argument('--batch_size', type=int, default=8)
parser.add_argument('--patch_size', type=int, default=256)
parser.add_argument('--val_batch_size', type=int, default=1)
parser.add_argument('--in_channels', type=int, default=3)
parser.add_argument('--model_name', type=str, default='Base2')
parser.add_argument('--test_name', type=str, default='val')
# 分布式训练
parser.add_argument('--dist_url', default='env://')
parser.add_argument('--world_size', default=1, type=int)
parser.add_argument('--local_rank', type=int, default=0)
parser.add_argument('--save_result', action='store_true')
parser.add_argument('--file_name', type=str, default='')
args_config, remaining = config_parser.parse_known_args()
if args_config.config:
with open(args_config.config, 'r') as f:
cfg = yaml.safe_load(f)
parser.set_defaults(**cfg)
args = parser.parse_args(remaining)
# if get_gpu_info.get_memory(num_gpu=args.num_gpus) is False:
# print('Out of the memory')
# while True:
# i = 999 * 9132877
# else:
# os.environ['CUDA_VISIBLE_DEVICES'] = ','.join(get_gpu_info.get_memory(num_gpu=args.num_gpus))
set_random_seed(args.seed)
init_distributed_mode(args)
set_config(args)