From bc25e17bed586148ff43008d0e2f9017297b258e Mon Sep 17 00:00:00 2001 From: "taehoon.kim" Date: Mon, 22 Mar 2021 16:15:12 +0900 Subject: [PATCH] post quantization using graph_fx(beta) --- post_quantization_validate.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/post_quantization_validate.py b/post_quantization_validate.py index 73c09dd4..f026d5ab 100755 --- 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'] = ""