test_disentanglement.sh

#!/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