nmsl_fovea.py 232 Bytes
Newer Older
Nianchen Deng's avatar
sync    
Nianchen Deng committed
1
2
def update_config(config):
    # Net parameters
Nianchen Deng's avatar
Nianchen Deng committed
3
4
    config.net = 'nmsl'
    config.fc.update({
Nianchen Deng's avatar
sync    
Nianchen Deng committed
5
6
7
        'nf': 128,
        'n_layers': 4
    })
Nianchen Deng's avatar
Nianchen Deng committed
8
9
    config.sa.update({
        'sample_range': (1, 50),
Nianchen Deng's avatar
sync    
Nianchen Deng committed
10
11
12
13
        'n_samples': 32
    })