char-rnn/char-rnn-chinese
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix convert bug, use double instead of float

Browse Source
This commit is contained in:
Jeff Zhang 2015-08-10 10:45:26 +08:00
parent 94e0ee8f39
commit e9ad92ebfd
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
convert.lua
View File

@ -33,6 +33,6 @@ else
end end
checkpoint = torch.load(opt.load_model) checkpoint = torch.load(opt.load_model)
checkpoint.protos.rnn:float() checkpoint.protos.rnn:double()
checkpoint.protos.criterion:float() checkpoint.protos.criterion:double()
torch.save(opt.save_file, checkpoint) torch.save(opt.save_file, checkpoint)

2
sample.lua
View File

@ -80,7 +80,7 @@ local num_layers = checkpoint.opt.num_layers
current_state = {} current_state = {}
for L = 1,checkpoint.opt.num_layers do for L = 1,checkpoint.opt.num_layers do
-- c and h for all layers -- c and h for all layers
local h_init = torch.zeros(1, checkpoint.opt.rnn_size):float() local h_init = torch.zeros(1, checkpoint.opt.rnn_size):double()
if opt.gpuid >= 0 then h_init = h_init:cuda() end if opt.gpuid >= 0 then h_init = h_init:cuda() end
table.insert(current_state, h_init:clone()) table.insert(current_state, h_init:clone())
table.insert(current_state, h_init:clone()) table.insert(current_state, h_init:clone())