Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.
And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.
Feature Request in Layers: Store best model with save_best_only.
See original GitHub issueTensorFlow.js version
0.10.3
Browser version
Latest Chrome
Describe the problem or feature request
When training on a just a few samples as usually happens in the browser, training and validation loss tend to jump around a bit.
It would be great if we could automatically get the best model after training finishes.
This is possible in Keras with the ModelCheckpoint(save_best_only=True) callback and afterwards loading the weights from that saved file.
Alternatively, if I knew how to copy a model in its current state, it would be straightforward to write the logic for this in a onEpochEnd callback.
Issue Analytics
- State:
- Created 5 years ago
- Comments:7
Top Results From Across the Web
Top Related Medium Post
Top Related StackOverflow Question
Troubleshoot Live Code
Top Related Reddit Thread
Top Related Hackernoon Post
Top Related Tweet
Top Related Dev.to Post
Top Related Hashnode Post
Great suggestion! FYI, model copying is coming soon. Please follow https://github.com/tensorflow/tfjs-core/pull/1038 for guidance on that front.
Closing as stale. Please @mention us if this needs more attention.