diff --git a/tools/pnnx/src/load_torchscript.cpp b/tools/pnnx/src/load_torchscript.cpp index bd2ac4528..3a25e42e4 100644 --- a/tools/pnnx/src/load_torchscript.cpp +++ b/tools/pnnx/src/load_torchscript.cpp @@ -690,9 +690,8 @@ int load_torchscript(const std::string& ptpath, Graph& pnnx_graph, input_tensors2.push_back(t); } } - else + else if(input_shapes2.size() != 0) { - std::vector input_tensors2; for (size_t i = 0; i < input_shapes2.size(); i++) { const std::vector& shape = input_shapes2[i];