Philipp Hoellmer commited on
Commit
e58815d
·
1 Parent(s): 5a7776a

Use lazy storage

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