testLSTM.lua

require 'nn'
require 'cudnn'
require 'cunn'
require 'rnn'

matio = require 'matio'

print('Loading test data...')
TestData =matio.load('lstm_data/lstm_testData.mat','lstm_testData')
print('test data loaded!')

model = torch.load('Results/Net_lstm')

Net_pred = torch.Tensor(TestData:size(1),TestData:size(2),8)
batchSize = 64
local SizeData = TestData:size(2)
SizeBatch = math.floor(TestData:size(2)/batchSize)
timer = torch.Timer()
  local yt,x;
  local num = 1;
  for NumBatches=1,SizeBatch do	    
  	print(NumBatches .. '/' .. SizeBatch)
	    x = TestData[{{},{num,num+batchSize-1},{}}]:cuda()    
	    y = model:forward(x)	    
	    Net_pred[{{},{num,num+batchSize-1},{}}] = y:float()
	    num = num + batchSize
   end

   if(TestData:size(1)%batchSize ~= 0) then	    
	    x = TestData[{{},{num,TestData:size(2)},{}}]:cuda()
	    y = model:forward(x)
	    Net_pred[{{},{num,TestData:size(2)},{}}] = y:float()
	end

matio.save('Results/Net_LSTMpred.mat',Net_pred); 
print(timer:time().real)