15 lines
No EOL
240 B
Bash
15 lines
No EOL
240 B
Bash
#!/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 |