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