retrain_with_baseline_isic2024_resnet18_3.job

#!/bin/bash

#SBATCH --partition=gpu_h100
#SBATCH --gpus=1
#SBATCH --job-name=retrain_baseline_res18
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=16
#SBATCH --time=05:00:00
#SBATCH --output=slurm_outputs/retrain_with_baseline_isic2024_resnet18_3_%A.out

module purge
module load 2023

source venv/bin/activate

python src/retrain_with_baseline.py --model resnet18 --dataset isic2024 --num_epochs 50 --threshold 0.3