File size: 1,006 Bytes
2af0e94 75854b3 2af0e94 75854b3 2af0e94 75854b3 2af0e94 75854b3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | #!/bin/bash -l
#SBATCH --job-name=om-train
#SBATCH --account=AIRR-P51-DAWN-GPU
#SBATCH --partition=pvc9
#SBATCH --nodes=1
#SBATCH --gres=gpu:1
#SBATCH -n 1
#SBATCH --time=10:00:00
#SBATCH --output=Logs/train_%j.out
#SBATCH --error=Logs/train_%j.err
. /etc/profile.d/modules.sh
module purge
module load rhel9/default-dawn
source ~/miniconda3/etc/profile.d/conda.sh
conda activate ~/rds/rds-airr-p51-TWhPgQVLKbA/Env/pub_env/pytorch-xpu
# export CUDA_VISIBLE_DEVICES=3
# export CUDA_VISIBLE_DEVICES=1,3
# export CUDA_VISIBLE_DEVICES=1,2,3
# # python -u OM_train.py -C Config/config_lct.yaml
# nohup python -u OM_train.py -C Config/config_lct.yaml > train_log.txt 2>&1 &
# python -u OM_train_2modes.py -C Config/config_om.yaml
# nohup python -u OM_train_2modes.py -C Config/config_om.yaml > train_log.txt 2>&1 &
nohup python -u OM_train_3modes.py -C Config/config_om.yaml --dummy-samples 20 --batchsize 1 > train_log.txt 2>&1 &
# nohup python -u OM_train.py -C Config/config_om.yaml > train_log.txt 2>&1 & |