def update_config(config): # Net parameters config.net = 'nerf' config.n_pos_encode = 10 config.fc.update({ 'nf': 96, 'n_layers': 4 }) config.sa.update({ 'sample_range': (1, 50), 'n_samples': 16, 'perturb_sample': False })