[ Conv2d ] save & read weight from file
authorjijoong.moon <jijoong.moon@samsung.com>
Tue, 9 Jun 2020 11:12:01 +0000 (20:12 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Wed, 10 Jun 2020 05:20:22 +0000 (14:20 +0900)
This PR provides read & save Kenel and Bias

**Self evaluation:**
1. Build test:  [X]Passed [ ]Failed [ ]Skipped
2. Run test:  [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: jijoong.moon <jijoong.moon@samsung.com>
nntrainer/src/conv2d_layer.cpp

index 9d161ed52d7d2239eb56864cec8fb0d8dd25607c..c0e6feb9e245eb08210e607938c0954b88843516 100644 (file)
@@ -66,11 +66,15 @@ int Conv2DLayer::initialize(bool last) {
 }
 
 void Conv2DLayer::read(std::ifstream &file) {
-  // NYI
+  std::for_each(filters.begin(), filters.end(),
+                [&](Tensor &i) { i.read(file); });
+  std::for_each(bias.begin(), bias.end(), [&](Tensor &i) { i.read(file); });
 }
 
 void Conv2DLayer::save(std::ofstream &file) {
-  // NYI
+  std::for_each(filters.begin(), filters.end(),
+                [&](Tensor i) { i.save(file); });
+  std::for_each(bias.begin(), bias.end(), [&](Tensor i) { i.save(file); });
 }
 
 Tensor Conv2DLayer::forwarding(Tensor in, int &status) {