bgnet.py

def update_config(config):
    # Net parameters
    config.net = 'bgnet'
    config.n_pos_encode = 10
    config.fc.update({
        'nf': 128,
        'n_layers': 4
    })