Skip to content

Commit

Permalink
Restructure custom_data_cached.yaml
Browse files Browse the repository at this point in the history
Use custom_data.yaml as default and only specify things that change with caching (i.e. the transforms)
  • Loading branch information
NiklasGebauer authored Oct 19, 2024
1 parent 6a1c975 commit f53e847
Showing 1 changed file with 2 additions and 12 deletions.
14 changes: 2 additions & 12 deletions src/schnetpack_gschnet/configs/data/custom_data_cached.yaml
Original file line number Diff line number Diff line change
@@ -1,15 +1,5 @@
_target_: schnetpack_gschnet.data.GenerativeAtomsDataModule

data_workdir: ${globals.data_workdir} # is specified in yaml-file of the experiment
placement_cutoff: ${globals.placement_cutoff}
use_covalent_radii: ${globals.use_covalent_radii}
covalent_radius_factor: ${globals.covalent_radius_factor}
num_workers: 8

datapath: ??? # path to the custom data base
batch_size: ??? # number of molecules per batch for training
num_train: ??? # number of molecules used for training
num_val: ??? # number of molecules used for validation (rest is put into test split)
defaults:
- custom_data

transforms:
- _target_: schnetpack.transform.SubtractCenterOfMass
Expand Down

0 comments on commit f53e847

Please sign in to comment.