#!/bin/bash
#SBATCH --job-name=EHD_AI_S2
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=20
#SBATCH --time=2-00:00:00
#SBATCH --output=/home/tanpv/Progs/CLI/EHD/logs/ehd_ai_%j.out
#SBATCH --error=/home/tanpv/Progs/CLI/EHD/logs/ehd_ai_%j.err
set -euo pipefail
ROOT=/home/tanpv/Progs/CLI/EHD
PY=/home/hapt/miniconda3/bin/python3
cd "$ROOT"
mkdir -p logs outputs/AI_Results/parts
export OMP_NUM_THREADS=1 MKL_NUM_THREADS=1 OPENBLAS_NUM_THREADS=1 NUMEXPR_NUM_THREADS=1
$PY ehd_ai.py preflight
srun --nodes=1 --ntasks=1 --cpus-per-task=20 --exclusive $PY ehd_ai.py train