From 0fd4304884aea11dec8ad2ec136a16637c62397b Mon Sep 17 00:00:00 2001
From: Stefan Doerr <stefdoerr@gmail.com>
Date: Mon, 19 Feb 2024 16:33:31 +0100
Subject: [PATCH] update ANI-2x training configuration

---
 examples/TensorNet-ANI2X.yaml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/examples/TensorNet-ANI2X.yaml b/examples/TensorNet-ANI2X.yaml
index a82f1d446..a3c753042 100644
--- a/examples/TensorNet-ANI2X.yaml
+++ b/examples/TensorNet-ANI2X.yaml
@@ -39,7 +39,7 @@ num_rbf: 32
 num_workers: 4
 output_model: Scalar
 precision: 32
-prior_model: null
+remove_ref_energy: true
 rbf_type: expnorm
 redirect: false
 reduce_op: add
@@ -48,8 +48,8 @@ splits: null
 seed: 1
 standardize: false
 test_interval: -1
-test_size: null
-train_size: 0.8
+test_size: 0
+train_size: 0.9
 trainable_rbf: false
 val_size: 0.1
 weight_decay: 0.0