File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -273,6 +273,9 @@ train = dp.Optimizer{
273
273
opt .learningRate = math.max (opt .minLR , opt .learningRate )
274
274
if not opt .silent then
275
275
print (" learningRate" , opt .learningRate )
276
+ if opt .meanNorm then
277
+ print (" mean gradParam norm" , opt .meanNorm )
278
+ end
276
279
end
277
280
end
278
281
end ,
@@ -283,9 +286,6 @@ train = dp.Optimizer{
283
286
if opt .cutoffNorm > 0 then
284
287
local norm = model :gradParamClip (opt .cutoffNorm ) -- affects gradParams
285
288
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
289
289
end
290
290
model :updateGradParameters (opt .momentum ) -- affects gradParams
291
291
model :updateParameters (opt .learningRate ) -- affects params
You can’t perform that action at this time.
0 commit comments