msl_rgb_periph.py 325 Bytes
Newer Older
BobYeah's avatar
BobYeah committed
1
2
3
4
5
6
7
def update_config(config):
    # Dataset settings
    config.GRAY = False

    # Net parameters
    config.NET_TYPE = 'msl'
    config.N_ENCODE_DIM = 10
BobYeah's avatar
sync    
BobYeah committed
8
9
10
11
12
    config.FC_PARAMS.update({
        'nf': 64,
        'n_layers': 8
    })
    config.SAMPLE_PARAMS.update({
BobYeah's avatar
BobYeah committed
13
        'depth_range': (1, 50),
BobYeah's avatar
sync    
BobYeah committed
14
15
        'n_samples': 4
    })