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