We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 275e22c commit 3601e9cCopy full SHA for 3601e9c
train.py
@@ -89,9 +89,9 @@ def train(opt):
89
frac = (epoch - opt.learning_rate_decay_start) // opt.learning_rate_decay_every
90
decay_factor = opt.learning_rate_decay_rate ** frac
91
opt.current_lr = opt.learning_rate * decay_factor
92
- utils.set_lr(optimizer, opt.current_lr) # set the decayed rate
93
else:
94
opt.current_lr = opt.learning_rate
+ utils.set_lr(optimizer, opt.current_lr)
95
# Assign the scheduled sampling prob
96
if epoch > opt.scheduled_sampling_start and opt.scheduled_sampling_start >= 0:
97
frac = (epoch - opt.scheduled_sampling_start) // opt.scheduled_sampling_increase_every
0 commit comments