-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdota_launch.sh
executable file
·32 lines (27 loc) · 1.49 KB
/
dota_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
#!/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=dota_train
#SBATCH --output=dota_train-%J.out
# mail alert at start, end and abortion of execution
#SBATCH --mail-type=ALL
# send mail to this address
#SBATCH [email protected]
module load gcc python cuda/11.4 opencv/4.5.5
source ~/envs/detectron2/bin/activate
mkdir $SLURM_TMPDIR/DOTA
mkdir $SLURM_TMPDIR/DOTA/train
unzip -q ~/projects/def-mpederso/akhil135/data_Aerial/DOTA/DOTA-train.zip -d $SLURM_TMPDIR
cp -r $SLURM_TMPDIR/DOTA-train/images/ $SLURM_TMPDIR/DOTA/train
cp ~/projects/def-mpederso/akhil135/data_Aerial/DOTA/annotations_DOTA_train.json $SLURM_TMPDIR/DOTA/
mkdir $SLURM_TMPDIR/DOTA/val
unzip -q ~/projects/def-mpederso/akhil135/data_Aerial/DOTA/DOTA-val.zip -d $SLURM_TMPDIR
cp -r $SLURM_TMPDIR/DOTA-val/images/ $SLURM_TMPDIR/DOTA/val
cp ~/projects/def-mpederso/akhil135/data_Aerial/DOTA/annotations_DOTA_val.json $SLURM_TMPDIR/DOTA/
#python train_net.py --resume --num-gpus 1 --config-file configs/dota/Semi-Sup-RCNN-FPN.yaml OUTPUT_DIR ~/scratch/DroneSSOD/DOTA_SS_1
python train_net.py --resume --num-gpus 1 --config-file configs/dota/Semi-Sup-RCNN-FPN-CROP.yaml OUTPUT_DIR ~/scratch/DroneSSOD/DOTA_CROP_SS_1_06
#python train_net.py --resume --num-gpus 1 --config-file configs/Dota-RCNN-FPN-CROP.yaml OUTPUT_DIR ~/scratch/DroneSSOD/DOTA_CROP_10