[fix] Set input layer when given input layer is empty

 - Added input layer from the graph when given input layer is empty

Signed-off-by: Jihoon Lee <jhoon.it.lee@samsung.com>
Signed-off-by: hyeonseok lee <hs89.lee@samsung.com>

diff --git a/nntrainer/models/neuralnet.cpp b/nntrainer/models/neuralnet.cpp
index 7d1469f..6874ea9 100644 (file)
--- a/nntrainer/models/neuralnet.cpp
+++ b/nntrainer/models/neuralnet.cpp
@@ -124,6 +124,12 @@ int NeuralNetwork::compile() {
     rep.push_back(*iter);
   }
 
+  if (input_layers.empty()) {
+    if (!rep.empty()) {
+      input_layers.emplace_back(rep.front()->getName());
+    }
+  }
+
   std::vector<std::unique_ptr<GraphRealizer>> realizers;
 
   realizers.emplace_back(new PreviousInputRealizer(input_layers));