def update_config(config): # Net parameters
config.net = 'mslfast' config.n_pos_encode = 6 config.fc.update({
'nf': 256, 'n_layers': 8 })
config.sa.update({ 'sample_range': (0.5, 5),
'n_samples': 128, 'perturb_sample': False })