def update_config(config): # Net parameters
config.net = 'snerffast2'
config.encode_x = 6 #config.encode_d = 4
config.fc.update({
'nf': 256,
'n_layers': 4
})
config.sa.update({ 'sample_range': (1, 50), 'n_samples': 32,
'perturb_sample': False })