Spaces:
Sleeping
Sleeping
File size: 543 Bytes
c42fe7e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
#!/bin/bash
#SBATCH --account=def-ichiro
#SBATCH --gpus-per-node=1
#SBATCH --mem=125G
#SBATCH --cpus-per-task=8
#SBATCH --time=20:00:00
#SBATCH --output=variance_training_%j.out
module load cuda/12.2 python/3.11
source ~/env-py311/bin/activate
export PYTHONPATH=$(pwd):$(pwd)/basics:$(pwd)/training:$PYTHONPATH
echo "GPU Check:"
nvidia-smi
python scripts/train.py \
--config=configs/CantusSVS_variance.yaml \
--exp_name=regular_variance_v1 \
--pl_trainer.accelerator=gpu \
--pl_trainer.devices=1 \
--pl_trainer.precision=16-mixed
|