Skip to content

Commit 5f268c9

Browse files
rnn fix
1 parent 00e45e0 commit 5f268c9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/recurrentlanguagemodel.lua

+3-3
Original file line numberDiff line numberDiff line change
@@ -273,6 +273,9 @@ train = dp.Optimizer{
273273
opt.learningRate = math.max(opt.minLR, opt.learningRate)
274274
if not opt.silent then
275275
print("learningRate", opt.learningRate)
276+
if opt.meanNorm then
277+
print("mean gradParam norm", opt.meanNorm)
278+
end
276279
end
277280
end
278281
end,
@@ -283,9 +286,6 @@ train = dp.Optimizer{
283286
if opt.cutoffNorm > 0 then
284287
local norm = model:gradParamClip(opt.cutoffNorm) -- affects gradParams
285288
opt.meanNorm = opt.meanNorm and (opt.meanNorm*0.9 + norm*0.1) or norm
286-
if opt.lastEpoch < report.epoch and not opt.silent then
287-
print("mean gradParam norm", opt.meanNorm)
288-
end
289289
end
290290
model:updateGradParameters(opt.momentum) -- affects gradParams
291291
model:updateParameters(opt.learningRate) -- affects params

0 commit comments

Comments
 (0)