Skip to content

Commit f9d2612

Browse files
authored
fix qconfig import for pytorch 1.10 (#9899)
1 parent c3aa6e9 commit f9d2612

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/callbacks/quantization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
from pytorch_lightning.utilities.exceptions import MisconfigurationException
2828

2929
if _TORCH_GREATER_EQUAL_1_10:
30-
from torch.ao.quantization import QConfig
30+
from torch.ao.quantization.qconfig import QConfig
3131
else:
3232
from torch.quantization import QConfig
3333

0 commit comments

Comments
 (0)