#!/bin/bash
#SBATCH --job-name=ds_ai
#SBATCH --partition=workq
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=16
#SBATCH --time=12:00:00
#SBATCH --output=/home/tanpv/Progs/claude/HW_EC_10/logs/slurm_ai_%j.log
cd /home/tanpv/Progs/claude/HW_EC_10
/home/tanpv/.conda/envs/ehd/bin/python -u src/downscale_ai.py \
    --folds 5 --subsample 4000000 --out outputs/downscale/ai_lgbm.json