| #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 & |