From 5780ec0a8bea9333a542d7e8904220ddebfac2ee Mon Sep 17 00:00:00 2001 From: Jihoon Lee Date: Mon, 19 Oct 2020 20:03:09 +0900 Subject: [PATCH] [NN] call setBatchSize at init setBatchSize is not called at init which cause problem when calling layers from outside. This fixes the issue **Self evaluation:** 1. Build test: [X]Passed [ ]Failed [ ]Skipped 2. Run test: [X]Passed [ ]Failed [ ]Skipped Signed-off-by: Jihoon Lee --- nntrainer/src/neuralnet.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/nntrainer/src/neuralnet.cpp b/nntrainer/src/neuralnet.cpp index 3c5b50b..16ed50b 100644 --- a/nntrainer/src/neuralnet.cpp +++ b/nntrainer/src/neuralnet.cpp @@ -249,6 +249,7 @@ int NeuralNetwork::init() { ml_logd("layer name: %s", l->getName().c_str()); initialized = true; + setBatchSize(batch_size); return status; } -- 2.7.4