Lines Matching full:setup

62 parser.add_argument('setup', type=str, help='setup yaml file')
74 with open(args.setup, 'r') as f:
75     setup = yaml.load(f.read(), yaml.FullLoader)  variable
79 setup_name = 'setup.yml'
84 if not 'name' in setup['model']:
85     print(f'warning: did not find model entry in setup, using default PitchPostFilter')
88     model_name = setup['model']['name']
106 # add repo info to setup
111         setup['repo'] = dict()
119         setup['repo']['hash'] = hash
120         setup['repo']['urls'] = urls
121         setup['repo']['dirty'] = is_dirty
125 # dump setup
127     yaml.dump(setup, f)
142 batch_size      = setup['training']['batch_size']
143 epochs          = setup['training']['epochs']
144 lr              = setup['training']['lr']
145 lr_decay_factor = setup['training']['lr_decay_factor']
148 data_config = setup['data']
149 data = LPCNetVocodingDataset(setup['dataset'], **data_config)
152 if 'validation_dataset' in setup:
153     validation_data = LPCNetVocodingDataset(setup['validation_dataset'], **data_config)
162 model = model_dict[model_name](*setup['model']['args'], **setup['model']['kwargs'])
189 w_l1 = setup['training']['loss']['w_l1']
190 w_lm = setup['training']['loss']['w_lm']
191 w_slm = setup['training']['loss']['w_slm']
192 w_sc = setup['training']['loss']['w_sc']
193 w_logmel = setup['training']['loss']['w_logmel']
194 w_wsc = setup['training']['loss']['w_wsc']
195 w_xcorr = setup['training']['loss']['w_xcorr']
196 w_sxcorr = setup['training']['loss']['w_sxcorr']
197 w_l2 = setup['training']['loss']['w_l2']
233     'setup'         : setup,