Browse Source

add value check for QuantizationAwareTraining's param optimize_option

tags/v1.1.0
yuchaojie 5 years ago
parent
commit
17082f0487
1 changed files with 7 additions and 2 deletions
  1. +7
    -2
      mindspore/compression/quant/quantizer.py

+ 7
- 2
mindspore/compression/quant/quantizer.py View File

@@ -17,6 +17,8 @@
from abc import ABC, abstractmethod
from enum import Enum

from ..._checkparam import Validator

__all__ = ["OptimizeOption", "Quantizer"]


@@ -39,12 +41,15 @@ class Quantizer(ABC):
This class is an abstract class.

Args:
optimize_option (OptimizeOption, list or tuple): Specifies the quant algorithm and options. Default: None.
optimize_option (OptimizeOption, list or tuple): Specifies the quant algorithm and options. Default:
OptimizeOption.QAT.
"""
def __init__(self,
optimize_option=None):
optimize_option=OptimizeOption.QAT):
if not isinstance(optimize_option, list) and not isinstance(optimize_option, tuple):
optimize_option = [optimize_option]
for option in optimize_option:
option = Validator.check_isinstance("optimize_option", option, OptimizeOption)
self.optimize_option = optimize_option

@abstractmethod


Loading…
Cancel
Save