#!/bin/bash #SBATCH --time=02:00:00 #SBATCH --mem=32GB #SBATCH --gres gpu:1 module load v100 module load cuda module load mamba source activate test python DisentanglementBase.py conda deactivate