Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Port training to TF2 #283

Open
wants to merge 2 commits into
base: users/boomanaiden154/main.port-training-to-tf2
Choose a base branch
from

Conversation

boomanaiden154
Copy link
Collaborator

This patch ports the training class to TF2. Most of the class needs
little porting as the TrainingEpochStats class is already completely
eager. The partially_restore_from_checkpoint function did not updating
to use new TF APIs. Future patches will fix everything up to use proper
TF2 APIs, but for now this works. Some future work needs to be done when
it comes up in tests to restore the step value as TF2 no longer contains
global state.

Created using spr 1.3.4
boomanaiden154 added a commit to boomanaiden154/gematria that referenced this pull request Jan 4, 2025
This patch ports the training class to TF2. Most of the class needs
little porting as the TrainingEpochStats class is already completely
eager. The partially_restore_from_checkpoint function did not updating
to use new TF APIs. Future patches will fix everything up to use proper
TF2 APIs, but for now this works. Some future work needs to be done when
it comes up in tests to restore the step value as TF2 no longer contains
global state.

Pull Request: google#283
boomanaiden154 added a commit to boomanaiden154/gematria that referenced this pull request Jan 6, 2025
This patch ports the training class to TF2. Most of the class needs
little porting as the TrainingEpochStats class is already completely
eager. The partially_restore_from_checkpoint function did not updating
to use new TF APIs. Future patches will fix everything up to use proper
TF2 APIs, but for now this works. Some future work needs to be done when
it comes up in tests to restore the step value as TF2 no longer contains
global state.

Pull Request: google#283
Created using spr 1.3.4
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