Lines Matching full:setup
57 'setup' : 'setup.yml',
67 parser.add_argument('setup', type=str, help='setup yaml file')
80 with open(args.setup, 'r') as f:
81 setup = yaml.load(f.read(), yaml.FullLoader) variable
87 if 'sparsification' in setup['lpcnet']['config']:
88 for sp_job in setup['lpcnet']['config']['sparsification'].values():
91 setup['training']['lr'] = 1.0e-5
92 setup['training']['lr_decay_factor'] = 0.0
93 setup['training']['epochs'] = 1
102 setup_name = 'setup.yml'
107 if not 'model' in setup['lpcnet']:
108 print(f'warning: did not find model entry in setup, using default lpcnet')
111 model_name = setup['lpcnet']['model']
130 # add repo info to setup
135 setup['repo'] = dict()
143 setup['repo']['hash'] = hash
144 setup['repo']['urls'] = urls
145 setup['repo']['dirty'] = is_dirty
149 # dump setup
151 yaml.dump(setup, f)
162 batch_size = setup['training']['batch_size']
163 epochs = setup['training']['epochs']
164 lr = setup['training']['lr']
165 lr_decay_factor = setup['training']['lr_decay_factor']
168 lpcnet_config = setup['lpcnet']['config']
169 data = LPCNetDataset( setup['dataset'],
173 frames_per_sample=setup['training']['frames_per_sample'],
179 if 'validation_dataset' in setup:
180 validation_data = LPCNetDataset( setup['validation_dataset'],
184 frames_per_sample=setup['training']['frames_per_sample'],
196 model = model_dict[model_name](setup['lpcnet']['config'])
227 'setup' : setup,