nudt_dsp
/
pnna_sdk
9
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update infer.py

This commit is contained in:
gwg_xujiao 2024-08-05 14:04:24 +08:00
parent f61fde111a
commit 5b0fe52fb5
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
netrans/netrans_py/infer.py
View File

@ -79,11 +79,10 @@ def main():
netrans_path = os.environ['NETRANS_PATH']
# netrans = os.path.join(netrans_path, 'pnnacc')
quantize_type = sys.argv[2]
cla = creat_cla(netrans_path, network_name,quantize_type,True)
cla = creat_cla(netrans_path, network_name,quantize_type,False)
# 调用量化函数
func = infer(cla)
print(func.model_name)
func = Infer(cla)
func.inference_network()
# 定义数据集文件路径