|
|
@ -20,6 +20,8 @@ from collections import OrderedDict
|
|
|
|
from contextlib import suppress
|
|
|
|
from contextlib import suppress
|
|
|
|
|
|
|
|
|
|
|
|
import torch.quantization
|
|
|
|
import torch.quantization
|
|
|
|
|
|
|
|
import torch.quantization.quantize_fx as quantize_fx
|
|
|
|
|
|
|
|
import copy
|
|
|
|
#currently, quantization only runs on CPUs
|
|
|
|
#currently, quantization only runs on CPUs
|
|
|
|
os.environ['CUDA_VISIBLE_DEVICES'] = ""
|
|
|
|
os.environ['CUDA_VISIBLE_DEVICES'] = ""
|
|
|
|
|
|
|
|
|
|
|
|