-
Notifications
You must be signed in to change notification settings - Fork 0
/
slurm_launch.sh
43 lines (37 loc) · 2.44 KB
/
slurm_launch.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
32
33
34
35
36
37
38
39
40
41
42
43
#!/bin/bash
#SBATCH --time=11:50:00
#SBATCH --account=def-mpederso
#SBATCH --gres=gpu:1 # Number of GPUs (per node)
#SBATCH --mem=24G # memory (per node)
# set name of job
#SBATCH --cpus-per-task=4
#SBATCH --job-name=visdrone_train
#SBATCH --output=vis_train-%J.out
# mail alert at start, end and abortion of execution
#SBATCH --mail-type=ALL
# send mail to this address
#SBATCH --mail-user=akhilpm135@gmail.com
module load gcc python cuda/11.4 opencv/4.5.5
module load httpproxy
source ~/envs/detectron2/bin/activate
export COMET_DISABLE_AUTO_LOGGING=1
mkdir $SLURM_TMPDIR/VisDrone
mkdir $SLURM_TMPDIR/VisDrone/train
unzip -q ~/projects/def-mpederso/akhil135/data_Aerial/VisDrone/VisDrone2019-DET-train.zip -d $SLURM_TMPDIR
cp -r $SLURM_TMPDIR/VisDrone2019-DET-train/images/ $SLURM_TMPDIR/VisDrone/train
cp ~/projects/def-mpederso/akhil135/data_Aerial/VisDrone/annotations_VisDrone_train.json $SLURM_TMPDIR/VisDrone/
mkdir $SLURM_TMPDIR/VisDrone/val
unzip -q ~/projects/def-mpederso/akhil135/data_Aerial/VisDrone/VisDrone2019-DET-val.zip -d $SLURM_TMPDIR
cp -r $SLURM_TMPDIR/VisDrone2019-DET-val/images/ $SLURM_TMPDIR/VisDrone/val
cp ~/projects/def-mpederso/akhil135/data_Aerial/VisDrone/annotations_VisDrone_val.json $SLURM_TMPDIR/VisDrone/
mkdir $SLURM_TMPDIR/VisDrone/test
unzip -q ~/projects/def-mpederso/akhil135/data_Aerial/VisDrone/VisDrone2019-DET-test.zip -d $SLURM_TMPDIR
cp -r $SLURM_TMPDIR/VisDrone2019-DET-test/images/ $SLURM_TMPDIR/VisDrone/test
cp ~/projects/def-mpederso/akhil135/data_Aerial/VisDrone/annotations_VisDrone_test.json $SLURM_TMPDIR/VisDrone/
#python train_net.py --num-gpus 1 --config-file configs/Base-RCNN-FPN.yaml OUTPUT_DIR ~/scratch/DroneSSOD/FPN_1
#python train_net.py --resume --num-gpus 1 --config-file configs/visdrone/Semi-Sup-RCNN-FPN.yaml OUTPUT_DIR ~/scratch/DroneSSOD/Comet_SS
#python train_net.py --num-gpus 1 --config-file configs/RCNN-FPN-CROP.yaml OUTPUT_DIR ~/scratch/DroneSSOD/FPN_CROP_1
python train_net.py --resume --num-gpus 1 --config-file configs/visdrone/Semi-Sup-RCNN-FPN-CROP.yaml OUTPUT_DIR ~/scratch/DroneSSOD/Temp
#python train_fcos.py --resume --num-gpus 1 --config-file configs/FCOS-CROP.yaml OUTPUT_DIR ~/scratch/DroneSSOD/FCOS_CROP
#python train_qdet.py --resume --num-gpus 1 --config-file configs/query_det/QueryDet.yaml OUTPUT_DIR ~/scratch/DroneSSOD/QDet_10
#python train_qdet.py --resume --num-gpus 1 --config-file configs/query_det/QueryDet_SSOD.yaml OUTPUT_DIR ~/scratch/DroneSSOD/QDet_SS