Skip to content

Commit 49c7679

Browse files
JeremallochJeremy Malloch
and
Jeremy Malloch
authored
Fixed path in quantization classification_flow (#1939)
- Fixed an import path in the pytorch-quantization tool. Previously it was trying to import models directly, but all models are defined within models.classifcation, so it would always default to the torchvision versions. As a result, the residual branch of ResNets was not being quantized. Signed-off-by: Jeremy Malloch <[email protected]> Co-authored-by: Jeremy Malloch <[email protected]>
1 parent f4a8635 commit 49c7679

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/pytorch-quantization/examples/torchvision/classification_flow.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@
3939

4040
import onnxruntime
4141
import numpy as np
42-
import models
42+
import models.classification as models
4343

4444
from prettytable import PrettyTable
4545

0 commit comments

Comments
 (0)