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