Skip to content

Conversation

@rphly
Copy link

@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
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