Lines Matching full:setup

71 parser.add_argument('setup', type=str, help='setup yaml file')
84 with open(args.setup, 'r') as f:
85     setup = yaml.load(f.read(), yaml.FullLoader)  variable
89 setup_name = 'setup.yml'
94 if not 'name' in setup['model']:
95     print(f'warning: did not find model entry in setup, using default PitchPostFilter')
98     model_name = setup['model']['name']
116 # add repo info to setup
121         setup['repo'] = dict()
131         setup['repo']['hash'] = hash
132         setup['repo']['urls'] = urls
133         setup['repo']['dirty'] = is_dirty
137 # dump setup
139     yaml.dump(setup, f)
144     testsignal, features, periods, numbits = load_inference_data(args.testdata, **setup['data'])
158 batch_size      = setup['training']['batch_size']
159 epochs          = setup['training']['epochs']
160 lr              = setup['training']['lr']
161 lr_decay_factor = setup['training']['lr_decay_factor']
164 data_config = setup['data']
165 data = SilkEnhancementSet(setup['dataset'], **data_config)
168 if 'validation_dataset' in setup:
169     validation_data = SilkEnhancementSet(setup['validation_dataset'], **data_config)
178 model = model_dict[model_name](*setup['model']['args'], **setup['model']['kwargs'])
205 w_l1 = setup['training']['loss']['w_l1']
206 w_lm = setup['training']['loss']['w_lm']
207 w_slm = setup['training']['loss']['w_slm']
208 w_sc = setup['training']['loss']['w_sc']
209 w_logmel = setup['training']['loss']['w_logmel']
210 w_wsc = setup['training']['loss']['w_wsc']
211 w_xcorr = setup['training']['loss']['w_xcorr']
212 w_sxcorr = setup['training']['loss']['w_sxcorr']
213 w_l2 = setup['training']['loss']['w_l2']
249     'setup'         : setup,