forked from charlesxzb/model-zoo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
nntc.config.yaml
39 lines (33 loc) · 834 Bytes
/
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
---
name: yolov3_tiny
gops: 5.57
model: $(home)/yolov3-tiny_jit.pt
int8_model: $(workdir)/$(name)_bmnetp_deploy_int8_unique_top.prototxt
int8_weight: $(workdir)/$(name)_bmnetp.int8umodel
fp_compile_options:
python3 -m bmnetp
--model $(model)
--shapes [1,3,416,416]
--target $(target)
--enable_profile True
--opt 1
--cmp 1
time_only_cali:
python3 -m ufw.cali.cali_model
--model $(model)
--test_iterations 10
--net_name $(name)
--postprocess_and_calc_score_class none
--debug_cmd='use_fake_data_to_test;not_call_bmnetu'
--target=$(target)
--cali_iterations 20
--input_shapes=[1,3,416,416]
bmnetu_options:
--opt=2
--cmp=True
--target $(target)
--model $(int8_model)
--weight $(int8_weight)
--seed 42
--enable_profile=True
bmnetu_batch_sizes: [1, 4, 8, 16]