#!/bin/bash


#PBS -l gpus=1
#PBS -l walltime=03:00:00
#PBS -m abe

module load PyTorch/2.1.2-foss-2023a-CUDA-12.1.1
module load Optuna/3.5.0-foss-2023a

cd $PBS_O_WORKDIR

source training_env/bin/activate

python main_cnn.py --method train