CoreML conversion failure: PyTorch convert function for op 'silu_' not implemented.

[Jacobsolawetz]

CoreML conversion failure: PyTorch convert function for op 'silu_' not implemented.

It seems like my coreml tools are having an issue with registering the silu opp here - Has anyone else encountered this issue?

yolov5-coreml-tools/src/coreml_export/main.py

Line 26 in 61a7441

def silu(context, node):

[Jacobsolawetz]

Added this and seems to be registering now

@register_torch_op
def silu_(context, node):
    inputs = _get_inputs(context, node, expected=1)
    x = inputs[0]
    y = mb.sigmoid(x=x)
    z = mb.mul(x=x, y=y, name=node.name)
    context.add(z)

[Leon0402]

Sorry for the very late response. These sort of issues are because of a mismatch of the pyTorch version coreml supports and yolov5 uses. Registering missing operators yourself is one workaround for this. And as you quoted I already this this for silu.

Not sure what silu_ is or how it's different from silu, but what you did seems exactly right to me.

Perhaps the library should be upgraded to a newer coreml version and be tested against new yolov5 versions, but I don't have the time to this myself.