Skip to content

Fix checkpointing and early stopping errors - #164

Open
rphly wants to merge 2 commits into
yisol:mainfrom
rphly:main
Open

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

Conversation

@rphly

@rphly rphly commented Nov 21, 2024

Copy link
Copy Markdown

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

rphly commented Nov 21, 2024

Copy link
Copy Markdown
Author

@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