-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
45 lines (29 loc) · 1000 Bytes
/
main.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
from app.utils.config import load_config
from app.utils.logging import Logger
from app.ml_engine import MLEngine
import json
import argparse
def arg_parser():
parser = argparse.ArgumentParser()
parser.add_argument('-c', '--config', type=str, help='config')
parser.add_argument('-d', '--ddd', type=str, help='train_id')
args = parser.parse_args()
return args
if __name__ == '__main__':
'''
args = arg_parser()
config = json.loads(args.config)
# config = load_config('./configs/lstm_train.json')
logger = Logger('app')
ME = MLEngine(config['model'])
ME.run()
'''
args = arg_parser()
config = json.loads(args.config)
print('##############################################')
print(f'train_id: {args.ddd}')
print('##############################################')
# config = load_config('./configs/lstm_train.json')
logger = Logger('app')
ME = MLEngine(config['model'], args.ddd)
ME.run()