cyclical_annealing icon indicating copy to clipboard operation
cyclical_annealing copied to clipboard

control learning rate in train_text_cyc_ptb.py example

Open d1024choi opened this issue 3 years ago • 0 comments

Could you explain why you control learning rate in optimizer in the example?

tmp = float((epoch-1)%args.cycle)/args.cycle cur_lr = args.lr*0.5*(1+np.cos(tmp*np.pi)) for param_group in optimizer.param_groups: param_group['lr'] = cur_lr

d1024choi avatar Mar 24 '21 02:03 d1024choi