From 443691414a1ea6f09807440fcae21333e14cf287 Mon Sep 17 00:00:00 2001
From: Francesco Gatti <gattifrancesco@hotmail.it>
Date: Sun, 16 Feb 2020 17:21:58 +0100
Subject: [PATCH] LSTM ok

---
 src/LSTM.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/LSTM.cpp b/src/LSTM.cpp
index 44502fa..6ecf4fd 100644
--- a/src/LSTM.cpp
+++ b/src/LSTM.cpp
@@ -118,11 +118,12 @@ LSTM::LSTM( Network *net, int hiddensize, bool returnSeq, std::string fname_weig
         net->dataType, net->tensorFormat, 3, dim_w));
 
     // load params
+    std::cout<<"Reading weights: PARAMS="<<cudnn_params*2<<"\n";
     readBinaryFile(fname_weights, cudnn_params*2, &w_h, &w_ptr);
     // set forward and backward params
     wf_ptr = w_ptr;
     wb_ptr = w_ptr + cudnn_params;
-    std::cout<<"wf: "<<wf_ptr<<" wb "<<wb_ptr<<"\n"; 
+    //std::cout<<"wf: "<<wf_ptr<<" wb "<<wb_ptr<<"\n"; 
 
     // set output dim 
     output_dim = input_dim;
@@ -138,7 +139,7 @@ LSTM::LSTM( Network *net, int hiddensize, bool returnSeq, std::string fname_weig
     checkCuda( cudaMalloc(&dstData, output_dim.tot()*sizeof(dnnType)) );
 
     // used during inference
-    dataDim_t  one_output_dim = input_dim;
+    one_output_dim = input_dim;
     one_output_dim.c = stateSize;
     checkCuda( cudaMalloc(&srcF, input_dim.tot()*sizeof(dnnType)) );
     checkCuda( cudaMalloc(&srcB, input_dim.tot()*sizeof(dnnType)) );