index 4093998d680c756250f3f7bc7484b3b608e98f6b..e7af5fe5465913e0ff84fa0fe54a2b82ea2a1053 100644 (file)
@@ -97,10 +97,15 @@ int timlCNNConvWriteToFile(FILE *fp1, FILE *fp2, FILE *fp3, timlCNNLayer *layer,
}\r
\r
if (fp3 != NULL) {\r
- timlUtilFwrite(layer->convParams.kernelInc, sizeof(float), dim, fp3);\r
- timlUtilFwrite(layer->convParams.kernelGradAccum, sizeof(float), dim, fp3);\r
- timlUtilFwrite(layer->convParams.biasInc, sizeof(float), layer->channel, fp3);\r
- timlUtilFwrite(layer->convParams.biasGradAccum, sizeof(float), layer->channel, fp3);\r
+ timlUtilFwrite(layer->featureMap, sizeof(float), layer->maxBatchSize*layer->row*layer->col*layer->channel, fp3);\r
+ timlUtilFwrite(layer->convParams.prevFeatureMapReshape, sizeof(float), layer->convParams.inputFeatureMapChannel*layer->convParams.kernelRow*layer->convParams.kernelCol*layer->row*layer->col, fp3);\r
+ if (layer->allocatorLevel == Util_AllocatorLevel1) {\r
+ timlUtilFwrite(layer->delta, sizeof(float), layer->maxBatchSize*layer->row*layer->col*layer->channel, fp3);\r
+ timlUtilFwrite(layer->convParams.kernelInc, sizeof(float), dim, fp3);\r
+ timlUtilFwrite(layer->convParams.kernelGradAccum, sizeof(float), dim, fp3);\r
+ timlUtilFwrite(layer->convParams.biasInc, sizeof(float), layer->channel, fp3);\r
+ timlUtilFwrite(layer->convParams.biasGradAccum, sizeof(float), layer->channel, fp3);\r
+ }\r
}\r
\r
return err;\r