Add weights to MD17 example

1713175349 · May 10, 2021 · 87bea21 · 87bea21
1 parent 53a8fbb
commit 87bea21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/train_GN_MD17.sh b/examples/train_GN_MD17.sh
@@ -4,5 +4,5 @@ dev=1 # list CUDA devices to use
 logdir="logs" # log dir for storing checkpoints and metrics
 
 mkdir -p $logdir
-CUDA_VISIBLE_DEVICES=$dev python scripts/train.py --conf examples/graph-network.yaml --dataset MD17 --dataset-root ~/data/md17 --dataset-arg aspirin --standardize true --log-dir $logdir --redirect true &
+CUDA_VISIBLE_DEVICES=$dev python scripts/train.py --conf examples/graph-network.yaml --dataset MD17 --dataset-root ~/data/md17 --dataset-arg aspirin --energy-weight 0.05 --force-weight 0.95 --standardize true --log-dir $logdir --redirect true &
 echo "Starting training... (saving logs at $logdir/log)"