-
Notifications
You must be signed in to change notification settings - Fork 17
/
predict.sh
executable file
·25 lines (17 loc) · 891 Bytes
/
predict.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
#/bin/bash
data_dir=$1
rm -rf submissions
mkdir submissions
export CUDA_VISIBLE_DEVICES=0
echo '0_inception_resnet_v2'
python3 src/predict.py --data_dir $data_dir --net 'inception_resnet_v2' --checkpoint_dir 'models/0_inception_resnet_v2' --batch_size 10 --image_size 299
echo '1_inception_resnet_v2'
python3 src/predict.py --data_dir $data_dir --net 'inception_resnet_v2' --checkpoint_dir 'models/1_inception_resnet_v2' --batch_size 10 --image_size 299
echo '2_resnet_v2_152'
python3 src/predict.py --data_dir $data_dir --net 'resnet_v2_152' --checkpoint_dir 'models/2_resnet_v2_152' --batch_size 10 --image_size 299
for i in {0..9}
do
printf "\ninception_resnet_v2_fold_$i\n"
python3 src/predict.py --data_dir $data_dir --net 'inception_resnet_v2' --checkpoint_dir "models/inception_resnet_v2_fold$i" --batch_size 10 --image_size 299
done
python3 src/prepare_submission.py