mace-osaka24-large.sh

python3 /opt/src/new_mace_0729/mace/mace/cli/run_train.py \
        --name="${MODEL_NAME}" \
        --train_file="/dataset/train" \
        --valid_file="/dataset/val" \
        --test_file="/dataset/test" \
        --statistics_file="/dataset/statistics.json" \
        --loss='universal' \
        --energy_weight=1 \
        --forces_weight=10 \
        --compute_stress=True \
        --stress_weight=100 \
        --stress_key='stress' \
        --eval_interval=1 \
        --error_table='PerAtomMAE' \
        --model="ScaleShiftMACE" \
        --interaction_first="RealAgnosticResidualInteractionBlock" \
        --interaction="RealAgnosticResidualInteractionBlock" \
        --num_interactions=2 \
        --correlation=3 \
        --max_ell=3 \
        --r_max=4.5 \
        --max_L=2 \
        --num_channels=128 \
        --num_radial_basis=10 \
        --MLP_irreps="16x0e" \
        --scaling='rms_forces_scaling' \
        --num_workers=64 \
        --lr=0.005 \
        --weight_decay=1e-8 \
        --ema \
        --ema_decay=0.995 \
        --scheduler_patience=5 \
        --batch_size=16 \
        --valid_batch_size=32 \
        --max_num_epochs=200 \
        --patience=50 \
        --amsgrad \
        --device=cuda \
        --seed=1 \
        --clip_grad=100 \
        --keep_checkpoints \
        --save_cpu \
	--restart_latest \
        --log_dir="/mnt/logs/${MODEL_NAME}" \
        --model_dir="/mnt/models/${MODEL_NAME}" \
        --checkpoints_dir="/mnt/checkpoints/${MODEL_NAME}" \
        --results_dir="/mnt/results/${MODEL_NAME}" \
        --distributed >> ${RESULT_FILE}