zhaojinxi edited a comment on pull request #9884: URL: https://github.com/apache/tvm/pull/9884#issuecomment-1008755169
@masahi I just use the **yolov5 of official implementation** https://github.com/ultralytics/yolov5/issues/251, with the command: "python export.py --weights yolov5s.pt --img 640 --include torchscript". Then i get a torchscript model. compile this model: x = torch.randn([1, 3, 640, 640]) model = torch.jit.load('yolov5s.torchscript') shape_list = [("input", x.numpy().shape)] mod, params = relay.frontend.from_pytorch(model, shape_list) then will get a error: Check failed: (!checked_type.defined()) is false: Expected Array[RelayExpr], but got Array[index 1: Array] I set a beakpoint at line 3276: outputs[node_name] = _expr.Tuple(inputs). When node_name==82, inputs contains list. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
