Philipp Hoellmer commited on
Commit
87dcc2a
·
1 Parent(s): 37fc2a7

Use lazy storage

Browse files
EncDec-ODE-Gamma/train.yaml CHANGED
@@ -88,19 +88,19 @@ data:
88
  class_path: omg.datamodule.StructureDataset
89
  init_args:
90
  file_path: "data/mp_20/train.lmdb"
91
- lazy_storage: False
92
  niggli_reduce: True
93
  val_dataset:
94
  class_path: omg.datamodule.StructureDataset
95
  init_args:
96
  file_path: "data/mp_20/val.lmdb"
97
- lazy_storage: False
98
  niggli_reduce: True
99
  pred_dataset:
100
  class_path: omg.datamodule.StructureDataset
101
  init_args:
102
  file_path: "data/mp_20/test.lmdb"
103
- lazy_storage: False
104
  niggli_reduce: True
105
  batch_size: 1024
106
  num_workers: 4
 
88
  class_path: omg.datamodule.StructureDataset
89
  init_args:
90
  file_path: "data/mp_20/train.lmdb"
91
+ lazy_storage: True
92
  niggli_reduce: True
93
  val_dataset:
94
  class_path: omg.datamodule.StructureDataset
95
  init_args:
96
  file_path: "data/mp_20/val.lmdb"
97
+ lazy_storage: True
98
  niggli_reduce: True
99
  pred_dataset:
100
  class_path: omg.datamodule.StructureDataset
101
  init_args:
102
  file_path: "data/mp_20/test.lmdb"
103
+ lazy_storage: True
104
  niggli_reduce: True
105
  batch_size: 1024
106
  num_workers: 4
EncDec-SDE-Gamma/train.yaml CHANGED
@@ -88,19 +88,19 @@ data:
88
  class_path: omg.datamodule.StructureDataset
89
  init_args:
90
  file_path: "data/mp_20/train.lmdb"
91
- lazy_storage: False
92
  niggli_reduce: True
93
  val_dataset:
94
  class_path: omg.datamodule.StructureDataset
95
  init_args:
96
  file_path: "data/mp_20/val.lmdb"
97
- lazy_storage: False
98
  niggli_reduce: True
99
  pred_dataset:
100
  class_path: omg.datamodule.StructureDataset
101
  init_args:
102
  file_path: "data/mp_20/test.lmdb"
103
- lazy_storage: False
104
  niggli_reduce: True
105
  batch_size: 32
106
  num_workers: 4
 
88
  class_path: omg.datamodule.StructureDataset
89
  init_args:
90
  file_path: "data/mp_20/train.lmdb"
91
+ lazy_storage: True
92
  niggli_reduce: True
93
  val_dataset:
94
  class_path: omg.datamodule.StructureDataset
95
  init_args:
96
  file_path: "data/mp_20/val.lmdb"
97
+ lazy_storage: True
98
  niggli_reduce: True
99
  pred_dataset:
100
  class_path: omg.datamodule.StructureDataset
101
  init_args:
102
  file_path: "data/mp_20/test.lmdb"
103
+ lazy_storage: True
104
  niggli_reduce: True
105
  batch_size: 32
106
  num_workers: 4
Linear-ODE-Gamma/train.yaml CHANGED
@@ -75,19 +75,19 @@ data:
75
  class_path: omg.datamodule.StructureDataset
76
  init_args:
77
  file_path: "data/mp_20/train.lmdb"
78
- lazy_storage: False
79
  niggli_reduce: True
80
  val_dataset:
81
  class_path: omg.datamodule.StructureDataset
82
  init_args:
83
  file_path: "data/mp_20/val.lmdb"
84
- lazy_storage: False
85
  niggli_reduce: True
86
  pred_dataset:
87
  class_path: omg.datamodule.StructureDataset
88
  init_args:
89
  file_path: "data/mp_20/test.lmdb"
90
- lazy_storage: False
91
  niggli_reduce: True
92
  batch_size: 128
93
  num_workers: 4
 
75
  class_path: omg.datamodule.StructureDataset
76
  init_args:
77
  file_path: "data/mp_20/train.lmdb"
78
+ lazy_storage: True
79
  niggli_reduce: True
80
  val_dataset:
81
  class_path: omg.datamodule.StructureDataset
82
  init_args:
83
  file_path: "data/mp_20/val.lmdb"
84
+ lazy_storage: True
85
  niggli_reduce: True
86
  pred_dataset:
87
  class_path: omg.datamodule.StructureDataset
88
  init_args:
89
  file_path: "data/mp_20/test.lmdb"
90
+ lazy_storage: True
91
  niggli_reduce: True
92
  batch_size: 128
93
  num_workers: 4
Linear-ODE/train.yaml CHANGED
@@ -69,19 +69,19 @@ data:
69
  class_path: omg.datamodule.StructureDataset
70
  init_args:
71
  file_path: "data/mp_20/train.lmdb"
72
- lazy_storage: False
73
  niggli_reduce: False
74
  val_dataset:
75
  class_path: omg.datamodule.StructureDataset
76
  init_args:
77
  file_path: "data/mp_20/val.lmdb"
78
- lazy_storage: False
79
  niggli_reduce: False
80
  pred_dataset:
81
  class_path: omg.datamodule.StructureDataset
82
  init_args:
83
  file_path: "data/mp_20/test.lmdb"
84
- lazy_storage: False
85
  niggli_reduce: False
86
  batch_size: 1024
87
  num_workers: 4
 
69
  class_path: omg.datamodule.StructureDataset
70
  init_args:
71
  file_path: "data/mp_20/train.lmdb"
72
+ lazy_storage: True
73
  niggli_reduce: False
74
  val_dataset:
75
  class_path: omg.datamodule.StructureDataset
76
  init_args:
77
  file_path: "data/mp_20/val.lmdb"
78
+ lazy_storage: True
79
  niggli_reduce: False
80
  pred_dataset:
81
  class_path: omg.datamodule.StructureDataset
82
  init_args:
83
  file_path: "data/mp_20/test.lmdb"
84
+ lazy_storage: True
85
  niggli_reduce: False
86
  batch_size: 1024
87
  num_workers: 4
Linear-SDE-Gamma/train.yaml CHANGED
@@ -89,19 +89,19 @@ data:
89
  class_path: omg.datamodule.StructureDataset
90
  init_args:
91
  file_path: "data/mp_20/train.lmdb"
92
- lazy_storage: False
93
  niggli_reduce: False
94
  val_dataset:
95
  class_path: omg.datamodule.StructureDataset
96
  init_args:
97
  file_path: "data/mp_20/val.lmdb"
98
- lazy_storage: False
99
  niggli_reduce: False
100
  pred_dataset:
101
  class_path: omg.datamodule.StructureDataset
102
  init_args:
103
  file_path: "data/mp_20/test.lmdb"
104
- lazy_storage: False
105
  niggli_reduce: False
106
  batch_size: 256
107
  num_workers: 4
 
89
  class_path: omg.datamodule.StructureDataset
90
  init_args:
91
  file_path: "data/mp_20/train.lmdb"
92
+ lazy_storage: True
93
  niggli_reduce: False
94
  val_dataset:
95
  class_path: omg.datamodule.StructureDataset
96
  init_args:
97
  file_path: "data/mp_20/val.lmdb"
98
+ lazy_storage: True
99
  niggli_reduce: False
100
  pred_dataset:
101
  class_path: omg.datamodule.StructureDataset
102
  init_args:
103
  file_path: "data/mp_20/test.lmdb"
104
+ lazy_storage: True
105
  niggli_reduce: False
106
  batch_size: 256
107
  num_workers: 4
Trig-ODE-Gamma/train.yaml CHANGED
@@ -82,19 +82,19 @@ data:
82
  class_path: omg.datamodule.StructureDataset
83
  init_args:
84
  file_path: "data/mp_20/train.lmdb"
85
- lazy_storage: False
86
  niggli_reduce: True
87
  val_dataset:
88
  class_path: omg.datamodule.StructureDataset
89
  init_args:
90
  file_path: "data/mp_20/val.lmdb"
91
- lazy_storage: False
92
  niggli_reduce: True
93
  pred_dataset:
94
  class_path: omg.datamodule.StructureDataset
95
  init_args:
96
  file_path: "data/mp_20/test.lmdb"
97
- lazy_storage: False
98
  niggli_reduce: True
99
  batch_size: 32
100
  num_workers: 4
 
82
  class_path: omg.datamodule.StructureDataset
83
  init_args:
84
  file_path: "data/mp_20/train.lmdb"
85
+ lazy_storage: True
86
  niggli_reduce: True
87
  val_dataset:
88
  class_path: omg.datamodule.StructureDataset
89
  init_args:
90
  file_path: "data/mp_20/val.lmdb"
91
+ lazy_storage: True
92
  niggli_reduce: True
93
  pred_dataset:
94
  class_path: omg.datamodule.StructureDataset
95
  init_args:
96
  file_path: "data/mp_20/test.lmdb"
97
+ lazy_storage: True
98
  niggli_reduce: True
99
  batch_size: 32
100
  num_workers: 4
Trig-ODE/train.yaml CHANGED
@@ -85,19 +85,19 @@ data:
85
  class_path: omg.datamodule.StructureDataset
86
  init_args:
87
  file_path: "data/mp_20/train.lmdb"
88
- lazy_storage: False
89
  niggli_reduce: False
90
  val_dataset:
91
  class_path: omg.datamodule.StructureDataset
92
  init_args:
93
  file_path: "data/mp_20/val.lmdb"
94
- lazy_storage: False
95
  niggli_reduce: False
96
  pred_dataset:
97
  class_path: omg.datamodule.StructureDataset
98
  init_args:
99
  file_path: "data/mp_20/test.lmdb"
100
- lazy_storage: False
101
  niggli_reduce: False
102
  batch_size: 32
103
  num_workers: 4
 
85
  class_path: omg.datamodule.StructureDataset
86
  init_args:
87
  file_path: "data/mp_20/train.lmdb"
88
+ lazy_storage: True
89
  niggli_reduce: False
90
  val_dataset:
91
  class_path: omg.datamodule.StructureDataset
92
  init_args:
93
  file_path: "data/mp_20/val.lmdb"
94
+ lazy_storage: True
95
  niggli_reduce: False
96
  pred_dataset:
97
  class_path: omg.datamodule.StructureDataset
98
  init_args:
99
  file_path: "data/mp_20/test.lmdb"
100
+ lazy_storage: True
101
  niggli_reduce: False
102
  batch_size: 32
103
  num_workers: 4
Trig-SDE-Gamma/train.yaml CHANGED
@@ -79,19 +79,19 @@ data:
79
  class_path: omg.datamodule.StructureDataset
80
  init_args:
81
  file_path: "data/mp_20/train.lmdb"
82
- lazy_storage: False
83
  niggli_reduce: True
84
  val_dataset:
85
  class_path: omg.datamodule.StructureDataset
86
  init_args:
87
  file_path: "data/mp_20/val.lmdb"
88
- lazy_storage: False
89
  niggli_reduce: True
90
  pred_dataset:
91
  class_path: omg.datamodule.StructureDataset
92
  init_args:
93
  file_path: "data/mp_20/test.lmdb"
94
- lazy_storage: False
95
  niggli_reduce: True
96
  batch_size: 32
97
  num_workers: 4
 
79
  class_path: omg.datamodule.StructureDataset
80
  init_args:
81
  file_path: "data/mp_20/train.lmdb"
82
+ lazy_storage: True
83
  niggli_reduce: True
84
  val_dataset:
85
  class_path: omg.datamodule.StructureDataset
86
  init_args:
87
  file_path: "data/mp_20/val.lmdb"
88
+ lazy_storage: True
89
  niggli_reduce: True
90
  pred_dataset:
91
  class_path: omg.datamodule.StructureDataset
92
  init_args:
93
  file_path: "data/mp_20/test.lmdb"
94
+ lazy_storage: True
95
  niggli_reduce: True
96
  batch_size: 32
97
  num_workers: 4
VESBD-ODE/train.yaml CHANGED
@@ -89,19 +89,19 @@ data:
89
  class_path: omg.datamodule.StructureDataset
90
  init_args:
91
  file_path: "data/mp_20/train.lmdb"
92
- lazy_storage: False
93
  niggli_reduce: False
94
  val_dataset:
95
  class_path: omg.datamodule.StructureDataset
96
  init_args:
97
  file_path: "data/mp_20/val.lmdb"
98
- lazy_storage: False
99
  niggli_reduce: False
100
  pred_dataset:
101
  class_path: omg.datamodule.StructureDataset
102
  init_args:
103
  file_path: "data/mp_20/test.lmdb"
104
- lazy_storage: False
105
  niggli_reduce: False
106
  batch_size: 512
107
  num_workers: 4
 
89
  class_path: omg.datamodule.StructureDataset
90
  init_args:
91
  file_path: "data/mp_20/train.lmdb"
92
+ lazy_storage: True
93
  niggli_reduce: False
94
  val_dataset:
95
  class_path: omg.datamodule.StructureDataset
96
  init_args:
97
  file_path: "data/mp_20/val.lmdb"
98
+ lazy_storage: True
99
  niggli_reduce: False
100
  pred_dataset:
101
  class_path: omg.datamodule.StructureDataset
102
  init_args:
103
  file_path: "data/mp_20/test.lmdb"
104
+ lazy_storage: True
105
  niggli_reduce: False
106
  batch_size: 512
107
  num_workers: 4
VPSBD-ODE/train.yaml CHANGED
@@ -74,19 +74,19 @@ data:
74
  class_path: omg.datamodule.StructureDataset
75
  init_args:
76
  file_path: "data/mp_20/train.lmdb"
77
- lazy_storage: False
78
  niggli_reduce: True
79
  val_dataset:
80
  class_path: omg.datamodule.StructureDataset
81
  init_args:
82
  file_path: "data/mp_20/val.lmdb"
83
- lazy_storage: False
84
  niggli_reduce: True
85
  pred_dataset:
86
  class_path: omg.datamodule.StructureDataset
87
  init_args:
88
  file_path: "data/mp_20/test.lmdb"
89
- lazy_storage: False
90
  niggli_reduce: True
91
  batch_size: 64
92
  num_workers: 4
 
74
  class_path: omg.datamodule.StructureDataset
75
  init_args:
76
  file_path: "data/mp_20/train.lmdb"
77
+ lazy_storage: True
78
  niggli_reduce: True
79
  val_dataset:
80
  class_path: omg.datamodule.StructureDataset
81
  init_args:
82
  file_path: "data/mp_20/val.lmdb"
83
+ lazy_storage: True
84
  niggli_reduce: True
85
  pred_dataset:
86
  class_path: omg.datamodule.StructureDataset
87
  init_args:
88
  file_path: "data/mp_20/test.lmdb"
89
+ lazy_storage: True
90
  niggli_reduce: True
91
  batch_size: 64
92
  num_workers: 4
VPSBD-SDE/train.yaml CHANGED
@@ -84,19 +84,19 @@ data:
84
  class_path: omg.datamodule.StructureDataset
85
  init_args:
86
  file_path: "data/mp_20/train.lmdb"
87
- lazy_storage: False
88
  niggli_reduce: True
89
  val_dataset:
90
  class_path: omg.datamodule.StructureDataset
91
  init_args:
92
  file_path: "data/mp_20/val.lmdb"
93
- lazy_storage: False
94
  niggli_reduce: True
95
  pred_dataset:
96
  class_path: omg.datamodule.StructureDataset
97
  init_args:
98
  file_path: "data/mp_20/test.lmdb"
99
- lazy_storage: False
100
  niggli_reduce: True
101
  batch_size: 64
102
  num_workers: 4
 
84
  class_path: omg.datamodule.StructureDataset
85
  init_args:
86
  file_path: "data/mp_20/train.lmdb"
87
+ lazy_storage: True
88
  niggli_reduce: True
89
  val_dataset:
90
  class_path: omg.datamodule.StructureDataset
91
  init_args:
92
  file_path: "data/mp_20/val.lmdb"
93
+ lazy_storage: True
94
  niggli_reduce: True
95
  pred_dataset:
96
  class_path: omg.datamodule.StructureDataset
97
  init_args:
98
  file_path: "data/mp_20/test.lmdb"
99
+ lazy_storage: True
100
  niggli_reduce: True
101
  batch_size: 64
102
  num_workers: 4