mlcgmd/test.txt at main · BioMedAI-UCSC/mlcgmd · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
Data module :  {
    '_target_': 'graphwm.data.datamodule.PLDataModule',
    'datasets': {
        'train': {
            '_target_': 'graphwm.data.PolymerDataset',
            'split': '${oc.env:PROJECT_ROOT}/splits/chain/train.txt',
            'directory': '${data.dir}',
            'dilation': '${model.dilation}',
            'seq_len': '${model.seq_len}',
            'grouping': '${model.cg_level}',
            'traj_len': '${data.traj_len}',
            'recursive_metis': '${data.recursive_metis}'
        }, 'val': [{
            '_target_': 'graphwm.data.PolymerDataset',
            'split': '${oc.env:PROJECT_ROOT}/splits/chain/val.txt',
            'directory': '${data.dir}',
            'dilation': '${model.dilation}',
            'seq_len': '${model.seq_len}',
            'grouping': '${model.cg_level}',
            'traj_len': '${data.traj_len}',
            'recursive_metis': '${data.recursive_metis}'
        }],
        'test': [{
            '_target_': 'graphwm.data.PolymerDataset',
            'split': '${oc.env:PROJECT_ROOT}/splits/chain/val.txt',
            'directory': '${data.dir}',
            'dilation': '${model.dilation}',
            'seq_len': '${model.seq_len}',
            'grouping': '${model.cg_level}',
            'traj_len': '${data.traj_len}',
            'recursive_metis': '${data.recursive_metis}'
        }]
    }, 'num_workers': {
            'train': 16,
            'val': 16,
            'test': 16
    }, 'batch_size': {
        'train': '${data.batch_size}',
        'val': '${data.batch_size}',
        'test': '${data.batch_size}'
    }
}