IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[POT] Add quantization templates to wheel (#10557)

Browse Source
This commit is contained in:
Liubov Talamanova 2022-02-22 09:11:43 +03:00 committed by GitHub
parent 33062bef7a
commit 1891967ad3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tools/pot/setup.py
View File

@ -25,6 +25,11 @@ class InstallCmd(install):
# install requires
self.do_egg_install()
def_quant_path = os.path.join("configs", "default_quantization_template.json")
aa_quant_path = os.path.join("configs", "accuracy_aware_quantization_template.json")
copyfile(def_quant_path, os.path.join(self.install_purelib, prefix, "pot", def_quant_path))
copyfile(aa_quant_path, os.path.join(self.install_purelib, prefix, "pot", aa_quant_path))
version_txt = os.path.join(prefix, "pot", "version.txt")
if os.path.exists(version_txt):
copyfile(os.path.join(version_txt),