nablaDFT

1
# Dataset config for torch geometric nablaDFT
2
_target_: nablaDFT.dataset.PyGHamiltonianDataModule
3

4
root: ${root}
5
dataset_name: ${dataset_name}
6
train_size: 0.9
7
val_size: 0.1
8
seed: ${seed}
9
# Dataloader args
10
batch_size: ${batch_size}
11
num_workers: ${num_workers}
12
persistent_workers: True
13
pin_memory: True
14