ser_nn_spectrogram/main_crmodel.py at master · DaiDongyang/ser_nn_spectrogram · GitHub

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
import tensorflow as tf
import argparse

from utils import parser_util
from utils import cfg_process
from CRModel import CRModel
from CRModel import data_set
from CRModel import load_data
from CRModel import cr_model_run


def add_arguments(parser):
    """Build ArgumentParser"""
    parser.add_argument('--config_file', type=str, default='./CRModel/CRModel.yml',
                        help='config file about hparams')
    parser.add_argument('--config_name', type=str, default='default',
                        help='config name for hparam')
    parser.add_argument('--gpu', type=str, default='',
                        help='config for CUDA_VISIBLE_DEVICES')


# def main(unused_argv):
#     parser = argparse.ArgumentParser()
#     add_arguments(parser)
#     flags, unparsed = parser.parse_known_args()
#     print('config file:', flags.config_file)
#     print('config name:', flags.config_name)
#     yparams = cfg_process.YParams(flags.config_file, flags.config_name)
#     flags = vars(flags)
#     yparams = cr_model_run.CRHParamsPreprocessor(yparams, flags).preprocess()
#     print('id str:', yparams.id_str)
#     yparams.save()
#     model = CRModel.CRModel(yparams)
#     l_data = load_data.load_data(yparams)
#     d_set = data_set.DataSet(l_data, yparams)
#     crmodel_run = cr_model_run.CRModelRun(model)
#     crmodel_run.run(d_set)


def main2(unused_argv):
    parser = parser_util.MyArgumentParser()
    add_arguments(parser)
    argc, flags_dict = parser.parse_to_dict()
    yparams = cfg_process.YParams(argc.config_file, argc.config_name)
    # flags = vars(flags)
    yparams = cr_model_run.CRHParamsPreprocessor(yparams, flags_dict).preprocess()
    print('id str:', yparams.id_str)
    yparams.save()
    if yparams.is_crmodel2:
        model = CRModel.CRModel2(yparams)
    else:
        model = CRModel.CRModel(yparams)
    l_data = load_data.load_data(yparams)
    d_set = data_set.DataSet(l_data, yparams)
    crmodel_run = cr_model_run.CRModelRun(model)
    crmodel_run.run(d_set)


if __name__ == '__main__':
    tf.app.run(main=main2)
    # main2(None)