-
-
Notifications
You must be signed in to change notification settings - Fork 25
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
Cannot recover model saved with Checkpointer() #161
Comments
We'd need to use |
Have made a quick-and-dirty fix, by creating a callback function, which can be executed after each epoch. Model is brought to CPU prior to saving its state. Have tested with BSON, but it might work with JLD2 as well.
|
Cheers,
Checkpointer()
saves the entire trained model with BSON. For several models, I am able to recover them withBSON.@load modeladdress model
. I am facing a case I could not recover the saved checkpoint. Error message as indicated below. Any hint on how to recover the saved stuff?Anyway, it seems advisable to change the function from saving the model to saving the outcome of
Flux.setup(model)
.Thanks.
The text was updated successfully, but these errors were encountered: