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.

Sign up for GitHub

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

Jump to bottom

Create API for Saving and Loading Model Snapshot #72

Open

e5l opened this issue May 25, 2021 · 1 comment

Assignees

Labels

question

Member

e5l commented May 25, 2021

I want to try out different training mechanisms using save-restore-replay. Could you tell me if it's possible to make API to support such use-cases?

zaleslaw added the question label

Collaborator

zaleslaw commented May 27, 2021 •

edited

Loading

Today you have a few approaches to save the model and continue work with that

Save the model as TF graph and weights in txt format (example). The model could be used for inference only after loading as InferenceModel.
Save the model as JSON config and weights in txt format (example). The model could be fine-tuned, training could be continued. This approach works well in both cases: you trained your model in Keras or KotlinDL firstly. The checkpointing could be organized via a sequence of model saving in this format.

Agree, that we need probably the special checkpointing API.

zaleslaw self-assigned this

zaleslaw added this to the 0.4 milestone

zaleslaw modified the milestones: 0.4, 0.5

ermolenkodev removed this from the 0.5 milestone

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment