snerffastx.py

def update_config(config):
    # Net parameters
    config.net = 'snerffastx4'
    config.n_pos_encode = 6
    #config.n_dir_encode = 4
    config.fc.update({
        'nf': 512,
        'n_layers': 8
    })
    config.sa.update({
        'sample_range': (0.3, 7),
        'n_samples': 128,
        'perturb_sample': False
    })