diff --git a/mindspore/lite/src/lite_model.cc b/mindspore/lite/src/lite_model.cc
index d6ba74b94a..5cb3fa8209 100644
--- a/mindspore/lite/src/lite_model.cc
+++ b/mindspore/lite/src/lite_model.cc
@@ -23,6 +23,7 @@
 #include <unordered_map>
 #include <memory>
 #include "src/common/prim_util.h"
+#include "src/common/graph_util.h"
 #ifdef ENABLE_V0
 #include "src/ops/compat/compat_register.h"
 #endif
@@ -219,6 +220,12 @@ int LiteModel::SubGraphVerify() const {
   auto tensor_size = this->all_tensors_.size();
   auto node_size = this->all_nodes_.size();
 
+  if (sub_graphs_[0]->input_indices_.size() == 0 || GetGraphInputNodes(this).size() == 0 ||
+      sub_graphs_[0]->output_indices_.size() == 0 || GetGraphOutputNodes(this).size() == 0) {
+    MS_LOG(ERROR) << "The model has invalid input and output, please check";
+    return RET_ERROR;
+  }
+
   for (auto &graph : this->sub_graphs_) {
     if (graph == nullptr) {
       MS_LOG(ERROR) << "graph is null.";