leon 1 month ago
parent
commit
097de83e22
1 changed files with 6 additions and 1 deletions
  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)
         {