File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 104104 "provider_options" : {
105105 "htp_performance_mode" : " burst" ,
106106 "htp_graph_finalization_optimization_mode" : " 3" ,
107+ "offload_graph_io_quantization" : " 0" ,
107108 "soc_model" : " 60"
108109 },
109110 "weight_sharing" : false
Original file line number Diff line number Diff line change @@ -164,7 +164,7 @@ def sam_mask_decoder_load(model_name):
164164
165165def ve_generate_quant_data (num_samples ):
166166 p = Path (ModelConfig .data_dir )
167- if p .is_dir () and (len (p .glob ("*images.npz" )) >= num_samples ):
167+ if p .is_dir () and (len ([ f for f in p .glob ("*images.npz" )] ) >= num_samples ):
168168 return
169169
170170 processor = SamProcessor .from_pretrained (ModelConfig .model_name )
@@ -182,7 +182,7 @@ def ve_generate_quant_data(num_samples):
182182
183183def md_generate_quant_data (num_samples ):
184184 p = Path (ModelConfig .data_dir )
185- if p .is_dir () and (len (p .glob ("*points.npz" )) >= num_samples ):
185+ if p .is_dir () and (len ([ f for f in p .glob ("*points.npz" )] ) >= num_samples ):
186186 return
187187
188188 model = SamModel .from_pretrained (ModelConfig .model_name ).to (device )
You can’t perform that action at this time.
0 commit comments