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