-
Notifications
You must be signed in to change notification settings - Fork 7.8k
Description
I am trying c++ deployment using tools/deploy/export_model.py. After successful conversion of the model to .ts file I am unable to do the inference in c++. I am getting the following error
terminate called after throwing an instance of 'c10::Error'
what(): forward() Expected a value of type 'Tensor' for argument 'argument_1' but instead found type 'Tuple[Dict[str, Tensor]]'.
Position: 1
Declaration: forward(torch.detectron2.export.flatten.TracingAdapter self, Tensor argument_1) -> ((Tensor, Tensor, Tensor, Tensor, Tensor))
Exception raised from checkArg at ../aten/src/ATen/core/function_schema_inl.h:162 (most recent call first):
frame #0: c10::Error::Error(c10::SourceLocation, std::__cxx11::basic_string<char, std::char_traits, std::allocator >) + 0x69 (0x7fa15a5deb29 in /home//libtorch/lib/libc10.so)
frame #1: c10::detail::torchCheckFail(char const*, char const*, unsigned int, std::__cxx11::basic_string<char, std::char_traits, std::allocator > const&) + 0xd2 (0x7fa15a5dbab2 in /home//libtorch/lib/libc10.so)