#!/bin/bash
#SBATCH --job-name=exp_clvr
#SBATCH --partition=reinforcement
#SBATCH --gres=gpu:1
#SBATCH --qos=gpujobs_long
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=20


# Load any necessary modules or activate the virtual environment
# module load cuda/11.0

# Optional: Change to the directory where your script is located
source /system/apps/userenv/miniconda3/bin/activate
conda init bash
conda activate ocl

# Run your script
python slot_attention/train.py --yml exp_clevr.yml --gpu_list 0
