forked from charlesxzb/model-zoo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
dbnet.nntc.config.yaml
executable file
·62 lines (54 loc) · 1.63 KB
/
dbnet.nntc.config.yaml
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
60
61
62
---
name: dbnet
model: $(home)/inference
dataset_path: $(home)/data/icdar2015
output_path: $(root)/output/$(name)/inference
time: true
precision: true
gops: [5.255]
div_info: div=0.229:0.224:0.225
mean_info: mean_value=0.485:0.456:0.406
size_info: resize_h=736,resize_w=1280;scale=0.00392156862
fp_batch_sizes: [1]
fp_loops:
- build_env:
- BMCOMPILER_LAYER_DTYPE_MODE=
fp_outdir_template: "{}b.fp32.compilation"
BM1684X:
fp_compile_options:
python3 -m bmpaddle
--model=$(model)
--shapes [4,3,736,1280]
--net_name $(name)
--target BM1684X
--cmp=False
--enable_profile True
# time_only_cali:
# python3 -u -m ufw.cali.cali_model
# --net_name $(name)
# --model $(model)/inference.pdmodel
# --input_shapes [1,3,736,1280]
# --cali_image_preprocess '$(size_info),$(mean_info),$(div_info),bgr2rgb=1'
# --cali_image_path $(dataset_path)/ch4_quanti_data
# --debug_cmd='to_umodel_not_cmp;quant_1684X;not_suspend;not_call_bmnetu'
# --target=BM1684X
# --cali_iterations 1
# --test_iterations 1
# --input_names "x"
# --output_names "sigmoid_0.tmp_0"
# bmnetu_batch_sizes: [1]
# int8_model: $(workdir)/$(name)_bmpaddle_deploy_int8_unique_top.prototxt
# int8_weight: $(workdir)/$(name)_bmpaddle.int8umodel
# bmnetu_options:
# --model $(int8_model)
# --weight $(int8_weight)
# --net_name $(name)
# --target BM1684X
# --cmp=False
# --v=4
# --opt 2
# --output_as_fp32 sigmoid_0.tmp_0
# --seed=42
# --enable_profile=True
build_env:
- BMCOMPILER_EXPORT_GRAPH=1