aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--nerv/examples/ptb/main.lua6
-rw-r--r--nerv/examples/ptb/reader.lua4
2 files changed, 5 insertions, 5 deletions
diff --git a/nerv/examples/ptb/main.lua b/nerv/examples/ptb/main.lua
index 688716b..5d1a326 100644
--- a/nerv/examples/ptb/main.lua
+++ b/nerv/examples/ptb/main.lua
@@ -2,13 +2,13 @@ nerv.include('reader.lua')
nerv.include('select_linear.lua')
gconf = {
- chunk_size = 5,
+ chunk_size = 15,
dropout_rate = 0,
lrate = 1.5,
wcost = 1e-5,
- max_iter = 3,
+ max_iter = 35,
clip = 5,
- momentum = 0,
+ momentum = 0.9,
batch_size = 200,
test = true,
}
diff --git a/nerv/examples/ptb/reader.lua b/nerv/examples/ptb/reader.lua
index 70c0c97..76a78cf 100644
--- a/nerv/examples/ptb/reader.lua
+++ b/nerv/examples/ptb/reader.lua
@@ -32,8 +32,8 @@ end
function Reader:get_seq(input_file)
local f = io.open(input_file, 'r')
self.seq = {}
- -- while true do
- for i = 1, 26 do
+ while true do
+ -- for i = 1, 26 do
local seq = f:read()
if seq == nil then
break