diff --git a/onnxconverter_common/float16.py b/onnxconverter_common/float16.py index 18a1a0f..6a02a68 100644 --- a/onnxconverter_common/float16.py +++ b/onnxconverter_common/float16.py @@ -82,7 +82,7 @@ def make_value_info_from_tensor(tensor): 'FeatureVectorizer', 'Imputer', 'LabelEncoder', 'LinearClassifier', 'LinearRegressor', 'Normalizer', 'OneHotEncoder', 'SVMClassifier', 'SVMRegressor', 'Scaler', 'TreeEnsembleClassifier', 'TreeEnsembleRegressor', 'ZipMap', 'NonMaxSuppression', 'TopK', - 'RoiAlign', 'Resize', 'Range', 'CumSum', 'Min', 'Max', 'Upsample', 'InstanceNormalization'] + 'RoiAlign', 'Resize', 'Range', 'CumSum', 'Min', 'Max', 'Upsample'] def convert_float_to_float16(model, min_positive_val=1e-7, max_finite_val=1e4,