nengo-dl icon indicating copy to clipboard operation
nengo-dl copied to clipboard

Nengo version of ModelCheckpoint callback

Open bmorcos opened this issue 5 years ago • 0 comments

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.

bmorcos avatar Mar 01 '21 17:03 bmorcos