WavyChannel-RF-ThermoHydraulics / scripts /run_experiment.slurm
yushuang88's picture
Upload folder using huggingface_hub
c023008 verified
Raw History Blame Contribute Delete
494 Bytes
#!/bin/bash
#SBATCH --job-name=wavy-rf-public
#SBATCH --cpus-per-task=8
#SBATCH --mem=16G
#SBATCH --time=00:30:00
set -euo pipefail
cd "${SLURM_SUBMIT_DIR:?Submit from the model package root}"
export OMP_NUM_THREADS=1
export OPENBLAS_NUM_THREADS=1
export MKL_NUM_THREADS=1
export NUMEXPR_NUM_THREADS=1
export MPLBACKEND=Agg
python -u scripts/train.py --config config/config.yaml
python -u scripts/inference.py --config config/config.yaml
python -u scripts/result.py --config config/config.yaml