model/fine_tune.sh

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 &