def update_config(config): # Net parameters
config.net = 'nmsl' config.fc.update({
'nf': 128, 'n_layers': 4 })
config.sa.update({ 'sample_range': (1, 50),
'n_samples': 32 })