forked from GuangxingHan/FCT
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_deb_single_voc1.slurm
36 lines (24 loc) · 837 Bytes
/
run_deb_single_voc1.slurm
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
32
33
34
35
36
#!/bin/bash
###SBATCH --partition=gpu-a100
#SBATCH --partition=deeplearn
#SBATCH --qos=gpgpudeeplearn
#SBATCH --constraint=dlg4|dlg5
###SBATCH --partition=feit-gpu-a100
###SBATCH --qos=feit
#SBATCH --job-name="deb"
#SBATCH --account=punim1623
#SBATCH --time=0-00:10:00
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=2
#SBATCH --gres=gpu:2
### "ntasks-per-node" should have same value as "res=gpu:"
#SBATCH --cpus-per-task=1
#SBATCH --mem=60G
module purge
eval "$(conda shell.bash hook)"
conda activate fct
CUDA_VISIBLE_DEVICES=0,1,2,3 python3 faster_rcnn_train_net.py --num-gpus 2 --dist-url auto \
--config-file configs/fsod/single_branch_pretraining_pascalvoc_split1_pvt_v2_b2_li.yaml 2>&1 | tee log/single_branch_pretraining_pascalvoc_split1_pvt_v2_b2_li.txt
##Log this job's resource usage stats###
my-job-stats -a -n -s
##