mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

LSTM ok

Browse Source
This commit is contained in:
Francesco Gatti
2020-02-16 17:21:58 +01:00
parent 4b48a2f38c
commit cd1e1c1582
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/LSTM.cpp
+3 -2
View File
@@ -118,11 +118,12 @@ LSTM::LSTM( Network *net, int hiddensize, bool returnSeq, std::string fname_weig
net->dataType, net->tensorFormat, 3, dim_w)); net->dataType, net->tensorFormat, 3, dim_w));
// load params // load params
std::cout<<"Reading weights: PARAMS="<<cudnn_params*2<<"\n";
readBinaryFile(fname_weights, cudnn_params*2, &w_h, &w_ptr); readBinaryFile(fname_weights, cudnn_params*2, &w_h, &w_ptr);
// set forward and backward params // set forward and backward params
wf_ptr = w_ptr; wf_ptr = w_ptr;
wb_ptr = w_ptr + cudnn_params; 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 // set output dim
output_dim = input_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)) ); checkCuda( cudaMalloc(&dstData, output_dim.tot()*sizeof(dnnType)) );
// used during inference // used during inference
dataDim_t one_output_dim = input_dim; one_output_dim = input_dim;
one_output_dim.c = stateSize; one_output_dim.c = stateSize;
checkCuda( cudaMalloc(&srcF, input_dim.tot()*sizeof(dnnType)) ); checkCuda( cudaMalloc(&srcF, input_dim.tot()*sizeof(dnnType)) );
checkCuda( cudaMalloc(&srcB, input_dim.tot()*sizeof(dnnType)) ); checkCuda( cudaMalloc(&srcB, input_dim.tot()*sizeof(dnnType)) );