Skip to content

Commit

Permalink
Update custom_data.yaml
Browse files Browse the repository at this point in the history
Add "train_sampler_cls" argument (for compatibility with new updated schnetpack version)
Remove "centered" argument from config for BuildAtomsTrajectory transform
  • Loading branch information
NiklasGebauer authored Oct 19, 2024
1 parent e669ab9 commit 6a1c975
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/schnetpack_gschnet/configs/data/custom_data.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -13,16 +13,18 @@ num_val: ??? # number of molecules used for validation (rest is put into tes

transforms:
- _target_: schnetpack.transform.SubtractCenterOfMass
- _target_: schnetpack_gschnet.transform.OrderByDistanceToOrigin
- _target_: schnetpack_gschnet.transform.ConditionalGSchNetNeighborList
model_cutoff: ${globals.model_cutoff}
prediction_cutoff: ${globals.prediction_cutoff}
placement_cutoff: ${globals.placement_cutoff}
use_covalent_radii: ${globals.use_covalent_radii}
covalent_radius_factor: ${globals.covalent_radius_factor}
- _target_: schnetpack_gschnet.transform.BuildAtomsTrajectory
centered: True
origin_type: ${globals.origin_type}
focus_type: ${globals.focus_type}
stop_type: ${globals.stop_type}
draw_random_samples: ${globals.draw_random_samples}
- _target_: schnetpack.transform.CastTo32

train_sampler_cls: null

0 comments on commit 6a1c975

Please sign in to comment.