slurm_finetune.sh

#!/usr/bin/zsh

### Add basic configuration for job

#SBATCH --job-name=fine_tuning
#SBATCH --output=logs/fine_tuning_%j.log
#SBATCH --error=logs/fine_tuning_error_%j.log
#SBATCH --nodes=1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=5
#SBATCH --gres=gpu:1
#SBATCH --time=22:00:00


###------------------------------------------------------------------------------------------------------------------------------

### Run the project in work directory of the cluster (configure based on need!! 
### RWTH File System : https://help.itc.rwth-aachen.de/en/service/rhr4fjjutttf/article/da307ec2c60940b29bd42ac483fc3ea7/
cd $HPCWORK
cd codebud/fine-tuning
###------------------------------------------------------------------------------------------------------------------------------

### JOB SCRIPT RUN
module load GCCcore/.13.2.0
module load Python/3.11.5
module load CUDA

source ../../venvs/codebud/bin/activate
echo $VIRTUAL_ENV

python --version

python main_fine_tuning.py

module unload CUDA
module unload Python/3.11.5

deactivate
echo "Script ran successfully"