forked from z-bingo/kernel-prediction-networks-PyTorch
-
Notifications
You must be signed in to change notification settings - Fork 1
/
run.sh
46 lines (46 loc) · 1.7 KB
/
run.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
44
45
46
#!/bin/bash
job=$1
##SBATCH --nodelist=gn19
#SBATCH --gres=gpu:8
#SBATCH -N 1
#SBATCH -n 1
#SBATCH --ntasks-per-node=1
#SBATCH --exclusive
##SBATCH -o job${job}.o
##SBATCH -e job${job}.e
#SBATCH -t 0-12
/bin/bash
module load ROCm/4.1.0
module load GCC/10.2.0
module load OpenMPI/4.0.5
module load PyTorch/1.8.1
pip install configobj
pip install torchsummary
pip install tensorboardx
pip install scikit-image==0.16.2
mkdir /scratch/yz87
rm -r /scratch/yz87/models/
rm -r /scratch/yz87/test_images/
rm -r /scratch/yz87/eval_images/
#rm DeepVideoDeblurring_Dataset_Original_High_FPS_Videos.zip
#rm disclaimer.txt
#rm -r __MACOSX
#rm -r original_high_fps_videos
mkdir /scratch/yz87/test_images/
mkdir /scratch/yz87/eval_images/
cd /scratch/yz87
if [ -f "DeepVideoDeblurring_Dataset_Original_High_FPS_Videos.zip" ]; then
echo "Dataset downloaded"
else
wget http://www.cs.ubc.ca/labs/imager/tr/2017/DeepVideoDeblurring/DeepVideoDeblurring_Dataset_Original_High_FPS_Videos.zip
unzip DeepVideoDeblurring_Dataset_Original_High_FPS_Videos.zip
fi
cd /home/cl114/yz87/spc2022/kernel-prediction-networks-PyTorch/
python dataset_test.py
rm -r /scratch/yz87/test_images/.DS_Store/
mkdir /scratch/yz87/models
cd /home/cl114/yz87/spc2022/kernel-prediction-networks-PyTorch/
python train_eval_syn.py --cuda --config_file kpn_128/kpn_config-${job}.conf --train_dir /scratch/yz87/test_images/ --mGPU --restart
wait
#srun --exclusive --nodes 1 --ntasks 1 python train_eval_syn.py --cuda --config_file kpn_specs/kpn_config-6.conf --train_dir /scratch/yz87/test_images/ --mGPU --restart
#srun --exclusive --nodes 1 --ntasks 1 python train_eval_syn.py --cuda --config_file kpn_specs/kpn_config-3.conf --train_dir /scratch/yz87/test_images/ --mGPU