-
Notifications
You must be signed in to change notification settings - Fork 7
/
gschnet_template.yaml
31 lines (28 loc) · 1.13 KB
/
gschnet_template.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# @package _global_
defaults:
- override /globals: gschnet_globals
- override /data: custom_data
- override /run: gschnet_run
- override /task: gschnet_task
- override /trainer: gschnet_trainer
- override /callbacks: gschnet_callbacks
- override /model: gschnet
- override /model/conditioning: null
- _self_
globals:
name: ??? # name of the experiment (parent directory of globals.id)
id: ??? # name of the directory where results of the run are stored
model_cutoff: ??? # cutoff applied in SchNet during feature extraction
prediction_cutoff: ??? # cutoff to select which atoms are used for prediction
placement_cutoff: ??? # cutoff used in the slicing of training molecules
atom_types: ??? # list of atom types that appear in the data set
use_covalent_radii: True
covalent_radius_factor: 1.1
draw_random_samples: 10
data_workdir: null
cache_workdir: null
data:
datapath: ??? # path to the custom data base
num_train: ??? # number of molecules used for training
num_val: ??? # number of molecules used for validation (rest -> test)
batch_size: 10