1. Enable network state write/read

[ti-machine-learning/ti-machine-learning.git] / src / common / cnn / timlCNNLinearWriteToFile.c

diff --git a/src/common/cnn/timlCNNLinearWriteToFile.c b/src/common/cnn/timlCNNLinearWriteToFile.c
index 0342f9b2569bfa2e2e7091b9c673f1d80d0fe1ba..14f0e5c589bbd2fab88182baff90aa44bfcf4f94 100644 (file)
--- a/src/common/cnn/timlCNNLinearWriteToFile.c
+++ b/src/common/cnn/timlCNNLinearWriteToFile.c
@@ -84,10 +84,14 @@ int timlCNNLinearWriteToFile(FILE *fp1, FILE *fp2, FILE *fp3, timlCNNLayer *laye
    }
 
    if (fp3 != NULL) {
-      timlUtilFwrite(layer->linearParams.weightInc, sizeof(float), layer->linearParams.dim*layer->linearParams.prevDim, fp3);
-      timlUtilFwrite(layer->linearParams.weightGradAccum, sizeof(float), layer->linearParams.dim*layer->linearParams.prevDim, fp3);
-      timlUtilFwrite(layer->linearParams.biasInc, sizeof(float), layer->linearParams.dim, fp3);
-      timlUtilFwrite(layer->linearParams.biasGradAccum, sizeof(float), layer->linearParams.dim, fp3);
+      timlUtilFwrite(layer->featureMap, sizeof(float), layer->maxBatchSize*layer->row*layer->col*layer->channel, fp3);
+      if (layer->allocatorLevel == Util_AllocatorLevel1) {
+         timlUtilFwrite(layer->delta, sizeof(float), layer->maxBatchSize*layer->row*layer->col*layer->channel, fp3);
+         timlUtilFwrite(layer->linearParams.weightInc, sizeof(float), layer->linearParams.dim*layer->linearParams.prevDim, fp3);
+         timlUtilFwrite(layer->linearParams.weightGradAccum, sizeof(float), layer->linearParams.dim*layer->linearParams.prevDim, fp3);
+         timlUtilFwrite(layer->linearParams.biasInc, sizeof(float), layer->linearParams.dim, fp3);
+         timlUtilFwrite(layer->linearParams.biasGradAccum, sizeof(float), layer->linearParams.dim, fp3);
+      }
    }
 
    return err;