-
Notifications
You must be signed in to change notification settings - Fork 0
/
main_train.py
51 lines (38 loc) · 1.21 KB
/
main_train.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
#!/usr/bin/env python
# -- coding: utf-8 --
"""
Copyright (c) 2018. All rights reserved.
Created by C. L. Wang on 2018/4/18
参考:
NumPy FutureWarning
https://stackoverflow.com/questions/48340392/futurewarning-conversion-of-the-second-argument-of-issubdtype-from-float-to
"""
import numpy as np
import ssl
# Bugfix: 参考https://blog.csdn.net/sylmoon/article/details/78777770
ssl._create_default_https_context = ssl._create_unverified_context
from data_loaders.my_data_loader import MyDataLoader
from models.my_model import MyModel
from trainers.my_trainer import MyTrainer
from utils.config_utils import init_config
def main_train():
"""
训练模型
:return:
"""
print('[Info] 解析配置...')
config = init_config('configs/my_config.json')
np.random.seed(47) # 固定随机数
print('[Info] 构造网络...')
model = MyModel(config=config)
print('[Info] 加载数据...')
dl = MyDataLoader(input_size=model.input_size, config=config)
print('[Info] 训练网络...')
trainer = MyTrainer(
model=model.model,
data=dl.get_data_dict(),
config=config)
trainer.train()
print('[Info] 训练完成...')
if __name__ == '__main__':
main_train()