meshed-memory-transformer icon indicating copy to clipboard operation
meshed-memory-transformer copied to clipboard

Call scheduler.step() per epoch AND per batch?

Open XuMengyaAmy opened this issue 2 years ago • 0 comments

In your train_xe function, you call scheduler.step() per epoch, then also call scheduler.step() per batch again. Is this your expected training strategy or an accidental mistake?

XuMengyaAmy avatar Dec 07 '21 12:12 XuMengyaAmy