File size: 628 Bytes
c2334a7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | nohup bash -c 'export CUDA_VISIBLE_DEVICES=0 && python -m mace.cli.run_train \
--name="MACE" \
--foundation_model="MACE-matpes-pbe-omat-ft.model" \
--multiheads_finetuning=False \
--train_file="train.extxyz" \
--test_file="test.extxyz" \
--valid_fraction=0.10 \
--energy_weight=1.0 \
--forces_weight=5.0 \
--E0s="average" \
--lr=0.001 \
--scaling="rms_forces_scaling" \
--batch_size=2 \
--max_num_epochs=1600 \
--ema \
--ema_decay=0.99 \
--amsgrad \
--default_dtype="float64" \
--device=cuda \
--seed=3 \
--restart_latest' > mace_train.log 2>&1 &
|