chamgj
/
netrans
forked from nudt_dsp/netrans
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix name confuse

This commit is contained in:
xujiao 2025-05-07 14:07:54 +08:00
parent 87e6c28404
commit dc12812dc6
26 changed files with 27 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
netrans_py/__pycache__/config.cpython-312.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/config.cpython-38.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/export.cpython-312.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/export.cpython-38.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/file_model.cpython-312.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/file_model.cpython-38.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/gen_inputmeta.cpython-312.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/import_model.cpython-312.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/import_model.cpython-38.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/netrans.cpython-312.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/netrans.cpython-38.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/quantize.cpython-312.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/quantize.cpython-38.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/utils.cpython-312.pyc Normal file
View File

Binary file not shown.

BIN
netrans_py/__pycache__/utils.cpython-38.pyc Normal file
View File

Binary file not shown.

4
netrans_py/config.py
View File

@ -1,7 +1,7 @@
import os
import sys
from utils import check_path, AttributeCopier, creat_cla
from utils import check_path, AttributeCopier, create_cls
class Config(AttributeCopier):
def __init__(self, source_obj) -> None:
@ -29,7 +29,7 @@ def main():
network_name = sys.argv[1]
# 构建 netrans 可执行文件的路径
netrans_path =os.getenv('NETRANS_PATH')
cla = creat_cla(netrans_path, network_name)
cla = create_cls(netrans_path, network_name)
func = InputmetaGen(cla)
func.inputmeta_gen()

4
netrans_py/import_model.py
View File

@ -1,7 +1,7 @@
import os
import sys
import subprocess
from utils import check_path, AttributeCopier, creat_cla
from utils import check_path, AttributeCopier, create_cls
def check_status(result):
if result.returncode == 0:
@ -218,7 +218,7 @@ def main():
netrans_path = os.environ['NETRANS_PATH']
# netrans = os.path.join(netrans_path, 'pnnacc')
clas = creat_cla(netrans_path, network_name,verbose=False)
clas = create_cls(netrans_path, network_name,verbose=False)
func = ImportModel(clas)
func.import_network()
if __name__ == "__main__":

4
netrans_py/infer.py
View File

@ -1,7 +1,7 @@
import os
import sys
import subprocess
from utils import check_path, AttributeCopier, creat_cla
from utils import check_path, AttributeCopier, create_cls
class Infer(AttributeCopier):
def __init__(self, source_obj) -> None:
@ -79,7 +79,7 @@ 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,False)
cla = create_cls(netrans_path, network_name,quantize_type,False)
# 调用量化函数
func = Infer(cla)

6
netrans_py/netrans.egg-info/PKG-INFO Normal file
View File

@ -0,0 +1,6 @@
Metadata-Version: 2.1
Name: netrans
Version: 0.1.0
Home-page: https://gitlink.org.cn/gwg_xujiao/netrans
Author: nudt_dsp
Requires-Dist: ruamel.yaml==0.18.6

7
netrans_py/netrans.egg-info/SOURCES.txt Normal file
View File

@ -0,0 +1,7 @@
README.md
setup.py
netrans.egg-info/PKG-INFO
netrans.egg-info/SOURCES.txt
netrans.egg-info/dependency_links.txt
netrans.egg-info/requires.txt
netrans.egg-info/top_level.txt

1
netrans_py/netrans.egg-info/dependency_links.txt Normal file
View File

@ -0,0 +1 @@

1
netrans_py/netrans.egg-info/requires.txt Normal file
View File

@ -0,0 +1 @@
ruamel.yaml==0.18.6

1
netrans_py/netrans.egg-info/top_level.txt Normal file
View File

@ -0,0 +1 @@

4
netrans_py/quantize.py
View File

@ -1,6 +1,6 @@
import os
import sys
from utils import check_path, AttributeCopier, creat_cla
from utils import check_path, AttributeCopier, create_cls
class Quantize(AttributeCopier):
def __init__(self, source_obj) -> None:
@ -83,7 +83,7 @@ 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)
cla = create_cls(netrans_path, network_name,quantize_type)
# 调用量化函数
run = Quantize(cla)

4
netrans_py/quantize_hb.py
View File

@ -1,6 +1,6 @@
import os
import sys
from utils import check_path, AttributeCopier, creat_cla
from utils import check_path, AttributeCopier, create_cls
class Quantize(AttributeCopier):
def __init__(self, source_obj) -> None:
@ -81,7 +81,7 @@ 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)
cla = create_cls(netrans_path, network_name,quantize_type)
# 调用量化函数
run = Quantize(cla)

2
netrans_py/utils.py
View File

@ -63,7 +63,7 @@ class AttributeCopier:
def _get_attribute_names(source_obj):
return source_obj.__dict__.keys()
class creat_cla(): #dataclass @netrans_params
class create_cls(): #dataclass @netrans_params
def __init__(self, netrans_path, name, quantized_type = 'uint8',verbose=False) -> None:
self.netrans_path = netrans_path
self.netrans = os.path.join(self.netrans_path, 'pnnacc')