nerf_periph.py 312 Bytes
Newer Older
BobYeah's avatar
BobYeah committed
1
2
def update_config(config):
    # Net parameters
Nianchen Deng's avatar
sync    
Nianchen Deng committed
3
    config.NET_TYPE = 'nerf'
BobYeah's avatar
BobYeah committed
4
    config.N_ENCODE_DIM = 10
BobYeah's avatar
sync    
BobYeah committed
5
    config.FC_PARAMS.update({
Nianchen Deng's avatar
sync    
Nianchen Deng committed
6
        'nf': 96,
BobYeah's avatar
sync    
BobYeah committed
7
8
9
        'n_layers': 4
    })
    config.SAMPLE_PARAMS.update({
BobYeah's avatar
BobYeah committed
10
        'depth_range': (1, 50),
Nianchen Deng's avatar
sync    
Nianchen Deng committed
11
12
        'n_samples': 16,
        'perturb_sample': False
BobYeah's avatar
sync    
BobYeah committed
13
    })