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

Nengo version of ModelCheckpoint callback #209

Open
bmorcos opened this issue Mar 1, 2021 · 0 comments
Open

Nengo version of ModelCheckpoint callback #209

bmorcos opened this issue Mar 1, 2021 · 0 comments

Comments

@bmorcos
Copy link

bmorcos commented Mar 1, 2021

Right now it's possible to use tf.keras.callbacks.ModelCheckpoint to save (keras) model weights during training under NengoDL, and then use sim.keras_model.load_weights(...) afterwards. However it might be nice to have our own version of this that uses the native save_params and load_params directly.

I briefly looked at the ModelCheckpoint code and after a quick scan it looks like it should be pretty reasonable to subclass.

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

No branches or pull requests

1 participant