pytorch 不能导出yolov5的onnx

    if opname == 'hardswish':
        relu=_registry[(domain, version)]["relu"]
        _registry[(domain, version)][opname] = relu

你可能感兴趣的:(AI,pytorch)