-
Notifications
You must be signed in to change notification settings - Fork 4
/
action40_config.py
31 lines (27 loc) · 960 Bytes
/
action40_config.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
import os
def check_dir(file_path):
if not os.path.exists(file_path):
os.mkdir(file_path)
class config():
def __init__(self):
self.py_dir = os.getcwd()
self.data_dir = os.path.join(self.py_dir,'data')
check_dir(self.data_dir)
self.save_ckpt_dir = os.path.join(self.py_dir,'ckpt')
check_dir(self.save_ckpt_dir)
self.class_info_dir = os.path.join(self.data_dir,'class.json')
self.best_ckpt_dir = os.path.join(self.save_ckpt_dir,'model_best.pth')
self.height = 224
self.width = 224
self.init_random_fc8 = True
self.learning_rate = 1e-5
self.max_iter_number = 100000
self.validate_iter_number = 500
self.save_ckpoints_iter_number = 10000
self.weight_decay = 5e-5
self.momentum = 0.99
self.train = False
if self.train:
self.batch_size = 8
else:
self.batch_size = 1