d1e6e624ef891543c5bba32bb0a696d350714693,coremltools/models/utils.py,,_convert_nn_spec_to_half_precision,#Any#,172

Before Change

            continue

        elif layer_type in quantized_layers:
            raise Exception("Half precision for " + layer_type +
                            " not yet implemented\n")
        else:
            raise Exception("Unknown layer " + layer_type)

    return spec

After Change

        if layer_type not in quantized_layers:
            params = getattr(layer, layer_type, None)
            params = params.ListFields() if params else []
            param_types = [p[0].message_type.name if p[0].message_type else None for p in params]

            if "WeightParams" in param_types:
                raise NotImplementedError("Quantization for layer "" + layer_type + "" not implemented.")
            continue  // print("Skipping layer {}. No need to quantize.".format(layer.name))

Instances