[KNN] Implement knn saving
authorJihoon Lee <jhoon.it.lee@samsung.com>
Thu, 11 Nov 2021 07:48:56 +0000 (16:48 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Wed, 17 Nov 2021 06:57:18 +0000 (15:57 +0900)
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 <jhoon.it.lee@samsung.com>
nntrainer/layers/centroid_knn.cpp
nntrainer/layers/centroid_knn.h

index f62f264..859722d 100644 (file)
@@ -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
index 200493b..018240f 100644 (file)
@@ -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()