nablaDFT

1
# Global variables
2
name: Equiformer_v2
3
dataset_name: dataset_train_tiny
4
max_steps: 1000000
5
warmup_steps: 0
6
job_type: train
7
pretrained: null # name of pretrained split or 'null'
8
ckpt_path: null # path to checkpoint for training resume or test run
9

10
# Datamodule parameters
11
root: ./datasets/nablaDFT/${.job_type}
12
batch_size: 2
13
num_workers: 4
14

15
# Devices
16
devices: [0]
17

18
# Trainer parameters
19
gradient_clip_algorithm: null
20
gradient_clip_val: null
21

22
# configs
23
defaults:
24
  - _self_
25
  - datamodule: nablaDFT_pyg.yaml  # dataset config
26
  - model: equiformer_v2_oc20.yaml  # model config
27
  - callbacks: default.yaml  # pl callbacks config
28
  - loggers: wandb.yaml  # pl loggers config
29
  - trainer: train.yaml  # trainer config
30

31
# need this to set working dir as current dir
32
hydra:
33
  output_subdir: null
34
  run:
35
    dir: .
36
original_work_dir: ${hydra:runtime.cwd}
37

38
seed: 23
39