Skip to content

Commit 12c09ce

Browse files
committed
fix: auto fetch tinynn
1 parent ea0ded7 commit 12c09ce

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

.gitignore

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,4 +63,7 @@ data/
6363
*.npy
6464

6565
# datasets
66-
datasets/
66+
datasets/
67+
68+
# quantization
69+
out/

tools/quantization.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,12 @@
99

1010
sys.path.insert(0, osp.dirname(osp.dirname(osp.abspath(__file__))))
1111

12+
try:
13+
import tinynn
14+
except ModuleNotFoundError:
15+
import subprocess
16+
subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'git+https://github.com/alibaba/TinyNeuralNetwork.git'])
17+
1218
from tinynn.graph.quantization.quantizer import QATQuantizer
1319
from tinynn.util.train_util import AverageMeter
1420
from tinynn.graph.tracer import model_tracer

0 commit comments

Comments
 (0)