gnh1201/Open-Llama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix epoch bug

Browse Source
This commit is contained in:
LiangSong 2023-05-06 09:45:37 +08:00
parent f893a0f5b8
commit 5b1f6a4861
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
solver/trainer.py
View File

@ -2,7 +2,7 @@
Author: LiangSong(sl12160010@gmail.com) Author: LiangSong(sl12160010@gmail.com)
Date: 2023-04-24 20:05:21 Date: 2023-04-24 20:05:21
LastEditors: LiangSong(sl12160010@gmail.com) LastEditors: LiangSong(sl12160010@gmail.com)
LastEditTime: 2023-05-04 08:41:37 LastEditTime: 2023-05-06 09:45:30
FilePath: /Open-Llama/solver/trainer.py FilePath: /Open-Llama/solver/trainer.py
Description: Description:
@ -172,6 +172,7 @@ class Trainer:
if self.data_step % self.save_interval == 0 and self.data_step > 0: if self.data_step % self.save_interval == 0 and self.data_step > 0:
self.accelerator.save_state(self.work_dir) self.accelerator.save_state(self.work_dir)
self.data_step += 1 self.data_step += 1
self.epoch += 1
wandb.finish() wandb.finish()
def log(self, losses): def log(self, losses):