-
Notifications
You must be signed in to change notification settings - Fork 0
/
baseline_engine_building_single_dnn.sh
executable file
·62 lines (47 loc) · 1.53 KB
/
baseline_engine_building_single_dnn.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
47
48
49
50
51
52
53
54
55
56
57
58
59
#!/usr/bin/env bash
mkdir baseline_engines
echo "\n\nStep 1: Building baseline engines\n"
echo "Experiment 1: ResNet152 Single DNN experiment"
#VGG-19 ResNet152 GPU & DLA
# python3 src/build_engine.py \
# --prototxt prototxt_input_files/resnet152.prototxt \
# --output baseline_engines/resnet152_gpu_only.plan \
# --start gpu \
# --transition -1 \
# --verbose
# #copied to use as a second engine for GPU use.
# cp baseline_engines/resnet152_gpu_only.plan baseline_engines/resnet152_gpu_only1.plan
# python3 src/build_engine.py \
# --prototxt prototxt_input_files/resnet152.prototxt \
# --output baseline_engines/resnet152_dla_only.plan \
# --start dla \
# --transition -1 \
# --verbose
# #resnet152_165_resnet152_364
# #Mensa
# python3 src/build_engine.py \
# --prototxt prototxt_input_files/resnet152.prototxt \
# --output baseline_engines/resnet152_dla_transition_at_165.plan \
# --start dla \
# --transition 165 \
# --verbose
# python3 src/build_engine.py \
# --prototxt prototxt_input_files/resnet152.prototxt \
# --output baseline_engines/resnet152_gpu_transition_at_364.plan \
# --start gpu \
# --transition 364 \
# --verbose
##HaX-CoNN
#resnet152_636_resnet152_165
python3 src/build_engine.py \
--prototxt prototxt_input_files/resnet152.prototxt \
--output baseline_engines/resnet152_dla_transition_at_636.plan \
--start dla \
--transition 635 \
--verbose
python3 src/build_engine.py \
--prototxt prototxt_input_files/resnet152.prototxt \
--output baseline_engines/resnet152_gpu_transition_at_165.plan \
--start gpu \
--transition 165 \
--verbose