中断训练后恢复训练时epoch计数不对

第二次中断训练后从save state继续第三次训练的时候，epoch计数会无视train_state.json里的记录数据并以第二次开始训练的数量作为第三次训练的起始计数， 并非从第一次开始算起的全局计数，不知道计数不对会不会对学习率的调度产生影响。