-
Notifications
You must be signed in to change notification settings - Fork 0
/
train_existing.sh
31 lines (29 loc) · 1023 Bytes
/
train_existing.sh
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
29
30
31
#!/bin/sh
### General options
### -- specify queue --
#BSUB -q gpuv100
#BSUB -R "select[gpu32gb]"
### -- set the job Name --
#BSUB -J t_existing
### -- ask for number of cores (default: 4) --
#BSUB -n 4
### -- Select the resources: 1 gpu in exclusive process mode --
#BSUB -gpu "num=1:mode=exclusive_process"
### -- specify that the cores must be on the same host --
#BSUB -R "span[hosts=1]"
### -- specify amount of memory per core/slot --
#BSUB -R "rusage[mem=5GB]"
### -- set walltime limit: hh:mm --
#BSUB -W 00:03
### -- send notification at start --
#BSUB -B
### -- send notification at completion --
#BSUB -N
### -- Specify the output and error file. %J is the job-id --
### -- -o and -e mean append, -oo and -eo mean overwrite --
#BSUB -o Logs/%J.out
#BSUB -e Logs/%J.err
module load python3/3.10.13
source torch-venv/bin/activate
# here follow the commands you want to execute
python -u AI/BGCactivityPrediction/CVAE_aa.py --test=True --existing_parameters=21550221_parameters --job_id=$LSB_JOBID