periph.py 318 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
    config.net = 'snerffast2'
Nianchen Deng's avatar
sync    
Nianchen Deng committed
4
5
    config.encode_x = 6
    #config.encode_d = 4
Nianchen Deng's avatar
Nianchen Deng committed
6
    config.fc.update({
Nianchen Deng's avatar
sync    
Nianchen Deng committed
7
        'nf': 256,
8
        'n_layers': 4
Nianchen Deng's avatar
sync    
Nianchen Deng committed
9
    })
Nianchen Deng's avatar
Nianchen Deng committed
10
11
12
    config.sa.update({
        'sample_range': (1, 50),
        'n_samples': 32,
Nianchen Deng's avatar
sync    
Nianchen Deng committed
13
14
        'perturb_sample': False
    })