Skip to content

Commit 0fd4304

Browse files
committed
update ANI-2x training configuration
1 parent 164343f commit 0fd4304

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/TensorNet-ANI2X.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ num_rbf: 32
3939
num_workers: 4
4040
output_model: Scalar
4141
precision: 32
42-
prior_model: null
42+
remove_ref_energy: true
4343
rbf_type: expnorm
4444
redirect: false
4545
reduce_op: add
@@ -48,8 +48,8 @@ splits: null
4848
seed: 1
4949
standardize: false
5050
test_interval: -1
51-
test_size: null
52-
train_size: 0.8
51+
test_size: 0
52+
train_size: 0.9
5353
trainable_rbf: false
5454
val_size: 0.1
5555
weight_decay: 0.0

0 commit comments

Comments
 (0)