From 8326301b1bfd5d5c0d2708270bb71e793b59dfce Mon Sep 17 00:00:00 2001 From: Jihoon Lee Date: Thu, 11 Nov 2021 16:48:56 +0900 Subject: [PATCH] [KNN] Implement knn saving As knn was not saving it's props, this patch implements saving property **Self evaluation:** 1. Build test: [X]Passed [ ]Failed [ ]Skipped 2. Run test: [X]Passed [ ]Failed [ ]Skipped Signed-off-by: Jihoon Lee --- nntrainer/layers/centroid_knn.cpp | 5 +++++ nntrainer/layers/centroid_knn.h | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/nntrainer/layers/centroid_knn.cpp b/nntrainer/layers/centroid_knn.cpp index f62f264..859722d 100644 --- a/nntrainer/layers/centroid_knn.cpp +++ b/nntrainer/layers/centroid_knn.cpp @@ -134,4 +134,9 @@ void CentroidKNN::calcDerivative(nntrainer::RunLayerContext &context) { throw std::invalid_argument("[CentroidKNN::calcDerivative] This Layer " "does not support backward propagation"); } + +void CentroidKNN::exportTo(nntrainer::Exporter &exporter, + const nntrainer::ExportMethods &method) const { + exporter.saveResult(centroid_knn_props, method, this); +} } // namespace nntrainer diff --git a/nntrainer/layers/centroid_knn.h b/nntrainer/layers/centroid_knn.h index 200493b..018240f 100644 --- a/nntrainer/layers/centroid_knn.h +++ b/nntrainer/layers/centroid_knn.h @@ -80,7 +80,7 @@ public: * @copydoc Layer::exportTo(Exporter &exporter, ExportMethods method) */ void exportTo(nntrainer::Exporter &exporter, - const nntrainer::ExportMethods &method) const override {} + const nntrainer::ExportMethods &method) const override; /** * @copydoc Layer::getType() -- 2.7.4