#!/bin/bash
#SBATCH --job-name=EHD_RARE_TRAIN
#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_rare_train_%j.out
#SBATCH --error=/home/tanpv/Progs/CLI/EHD/logs/ehd_rare_train_%j.err
set -euo pipefail
cd /home/tanpv/Progs/CLI/EHD
mkdir -p logs outputs/AI_RareEvent_Results/parts
export OMP_NUM_THREADS=1 MKL_NUM_THREADS=1 OPENBLAS_NUM_THREADS=1 NUMEXPR_NUM_THREADS=1
/home/hapt/miniconda3/bin/python3 ehd_rare_event.py preflight
srun --nodes=1 --ntasks=1 --cpus-per-task=20 --exclusive /home/hapt/miniconda3/bin/python3 ehd_rare_event.py train