Skip to content

Fix checkpointing and early stopping errors#164

Open
rphly wants to merge 2 commits intoyisol:mainfrom
rphly:main
Open

Fix checkpointing and early stopping errors#164
rphly wants to merge 2 commits intoyisol:mainfrom
rphly:main

Conversation

@rphly
Copy link
Copy Markdown

@rphly rphly commented Nov 21, 2024

created this pr so that epoch checkpointing works as intended.

also found another bug where global steps was being updated twice, once before and once after syncing gradients.
This causes +2 to the progress bar and also early stopping due to args.max_train_steps

@rphly
Copy link
Copy Markdown
Author

rphly commented Nov 21, 2024

@yisol

@rphly rphly changed the title fix epoch checkpointing Fix checkpointing and early stopping errors Nov 21, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant