[ahub] fix ahub issue

[platform/core/ml/nntrainer.git] / nntrainer / models / neuralnet.cpp

diff --git a/nntrainer/models/neuralnet.cpp b/nntrainer/models/neuralnet.cpp
index feb1284..a9148bf 100644 (file)
--- a/nntrainer/models/neuralnet.cpp
+++ b/nntrainer/models/neuralnet.cpp
@@ -407,7 +407,7 @@ void NeuralNetwork::save(const std::string &file_path,
     }
     if (opt && istrequal(opt->getType(), "adam")) {
       std::string adam = "adam";
-      model_file.write(adam.c_str(), adam.size());
+      model_file.write(adam.c_str(), 4);
       for (auto iter = model_graph.cbegin(); iter != model_graph.cend();
            iter++) {
         (*iter)->save(model_file, true);