forked from maups/hydra-fmow
-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathtrain.sh
More file actions
60 lines (41 loc) · 4.08 KB
/
train.sh
File metadata and controls
60 lines (41 loc) · 4.08 KB
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
47
48
49
50
51
52
53
54
55
56
57
58
59
#!/bin/bash
set -e
time=$(date +%s)
if [[ -d /wdata/working/cnn_checkpoint_weights/ && $(ls /wdata/working/cnn_checkpoint_weights/) != "" ]]
then
echo "REMOVING OLD MODEL FILES: $(ls /wdata/working/cnn_checkpoint_weights/)"
rm /wdata/working/cnn_checkpoint_weights/*
fi
if [[ -d /wdata/input/train_data/ && $(ls /wdata/input/train_data/) != "" ]]
then
echo "REMOVING OLD TRAIN FILES: $(ls /wdata/input/train_data/)"
rm -r /wdata/input/train_data/*
fi
echo "PREPROCESSING TRAIN DATASET"
python iarpa/runBaseline.py --prepare True --train True
echo "GENERATING MODEL 1/12"
python -u iarpa/runBaseline.py --algorithm resnet50 --train True --num_gpus 4 --num_epochs 12 --batch_size 80 --class_weights no_weights --prefix p01 --generator flip --database v1
echo "GENERATING MODEL 2/12"
python -u iarpa/runBaseline.py --algorithm densenet --train True --num_gpus 4 --num_epochs 12 --batch_size 28 --class_weights no_weights --prefix p01 --generator flip --database v1
echo "GENERATING MODEL 3/12"
python -u iarpa/runBaseline.py --algorithm densenet --train True --num_gpus 4 --num_epochs 5 --batch_size 28 --class_weights class_pond --prefix p02 --generator flip --fine_tunning True --load_weights /wdata/working/cnn_checkpoint_weights/weights.v1.densenet.p01.06.hdf5 --database v2
echo "GENERATING MODEL 4/12"
python -u iarpa/runBaseline.py --algorithm resnet50 --train True --num_gpus 4 --num_epochs 5 --batch_size 80 --fine_tunning True --class_weights class_pond --prefix p02 --generator flip --load_weights /wdata/working/cnn_checkpoint_weights/weights.v1.resnet50.p01.06.hdf5 --database v3
echo "GENERATING MODEL 5/12"
python -u iarpa/runBaseline.py --algorithm resnet50 --train True --num_gpus 4 --num_epochs 5 --batch_size 80 --fine_tunning True --class_weights class_pond --prefix p03 --generator flip --load_weights /wdata/working/cnn_checkpoint_weights/weights.v1.resnet50.p01.06.hdf5 --database v2
echo "GENERATING MODEL 6/12"
python -u iarpa/runBaseline.py --algorithm densenet --train True --num_gpus 4 --num_epochs 5 --batch_size 28 --class_weights class_pond --prefix p03 --generator flip --fine_tunning True --load_weights /wdata/working/cnn_checkpoint_weights/weights.v1.densenet.p01.06.hdf5 --database v3
echo "GENERATING MODEL 7/12"
python -u iarpa/runBaseline.py --algorithm densenet --train True --num_gpus 4 --num_epochs 5 --batch_size 28 --class_weights sklearn_class_weight --prefix p04 --generator zoom --load_weights /wdata/working/cnn_checkpoint_weights/weights.v1.densenet.p01.06.hdf5 --database v1
echo "GENERATING MODEL 8/12"
python -u iarpa/runBaseline.py --algorithm densenet --train True --num_gpus 4 --num_epochs 5 --batch_size 28 --class_weights no_weights --prefix p05 --generator shift --fine_tunning True --load_weights /wdata/working/cnn_checkpoint_weights/weights.v1.densenet.p01.06.hdf5 --database v1
echo "GENERATING MODEL 9/12"
python -u iarpa/runBaseline.py --algorithm densenet --train True --num_gpus 4 --num_epochs 5 --batch_size 28 --class_weights class_weights --prefix p06 --generator shift --fine_tunning True --load_weights /wdata/working/cnn_checkpoint_weights/weights.v1.densenet.p01.06.hdf5 --database v3
echo "GENERATING MODEL 10/12"
python -u iarpa/runBaseline.py --algorithm resnet50 --train True --num_gpus 4 --num_epochs 5 --batch_size 80 --fine_tunning True --class_weights sklearn_class_weight --prefix p04 --generator flip --load_weights /wdata/working/cnn_checkpoint_weights/weights.v1.resnet50.p01.06.hdf5 --database v3
echo "GENERATING MODEL 11/12"
python -u iarpa/runBaseline.py --algorithm densenet --train True --num_gpus 4 --num_epochs 4 --batch_size 28 --class_weights sklearn_class_weight --prefix p07 --generator flip --load_weights /wdata/working/cnn_checkpoint_weights/weights.v1.densenet.p01.06.hdf5 --database v1
echo "GENERATING MODEL 12/12"
python -u iarpa/runBaseline.py --algorithm densenet --train True --num_gpus 4 --num_epochs 5 --batch_size 28 --class_weights sklearn_class_weight --prefix p08 --generator flip --fine_tunning True --load_weights /wdata/working/cnn_checkpoint_weights/weights.v1.densenet.p01.06.hdf5 --database v2
echo "TRAINING FINISHED"
exit