def update_config(config): # Net parameters
config.net = 'snerffastx4'
config.encode_x = 6 #config.encode_d = 4
config.fc.update({
'nf': 512, 'n_layers': 8 })
config.sa.update({ 'sample_range': (0.3, 7),
'n_samples': 128, 'perturb_sample': False })