未验证 提交 31f08bdb 编写于 作者: H huangxu96 提交者: GitHub

remove quantized layers type check in paddleslime, does it in paddle instead. (#611)

Co-authored-by: NBai Yifan <me@ethanbai.com>
上级 7310387f
......@@ -33,8 +33,6 @@ BUILT_IN_PREPROCESS_TYPES = ['PACT']
VALID_DTYPES = ['int8']
DYGRAPH_QUANTIZABLE_TYPE = ['Conv2D', 'Linear']
__all__ = ['QAT']
_quant_config_default = {
......@@ -120,11 +118,6 @@ def _parse_configs(user_config):
assert isinstance(configs['quantizable_layer_type'], list), \
"quantizable_layer_type must be a list"
for op_type in configs['quantizable_layer_type']:
assert (op_type in DYGRAPH_QUANTIZABLE_TYPE), "{} is not support, \
now support op types are {}".format(
op_type, DYGRAPH_QUANTIZABLE_TYPE)
return configs
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册