post quantization using graph_fx(beta)

4 years ago · bc25e17bed
parent 2b2e9c77fd
commit bc25e17bed
1 changed files with 2 additions and 0 deletions
--- a/post_quantization_validate.py
+++ b/post_quantization_validate.py
@ -20,6 +20,8 @@ from collections import OrderedDict
 from contextlib import suppress

 import torch.quantization
+import torch.quantization.quantize_fx as quantize_fx
+import copy
 #currently, quantization only runs on CPUs
 os.environ['CUDA_VISIBLE_DEVICES'] = ""