leon 1 月之前
父節點
當前提交
097de83e22
共有 1 個文件被更改,包括 6 次插入1 次删除
  1. 6 1
      src/nodes/infer/inferNode.cpp

+ 6 - 1
src/nodes/infer/inferNode.cpp

@@ -52,7 +52,12 @@ void InferNode::work()
             {
                 printf("Unexpected result type from model");
                 throw std::runtime_error("Unexpected result type from model");
-            }            
+            }
+            for (auto& output_buffer : output_buffers_)
+            {
+                // printf("Node %s push data to %s\n", name_.c_str(), output_buffer.first.c_str());
+                output_buffer.second->push(metaData);
+            }           
         }
         if (!has_data)
         {