-
Notifications
You must be signed in to change notification settings - Fork 58
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
saving tensorflow models #11
Comments
Hi, Good question; I remember saving and loading model states to be a bit of a hassle in TensorFlow. |
Thanks for the info - I guess I was thinking more of how to store in the AIde db. The keras wrapper saves weights and architecture in hdf5 format but I couldn't find a way to save this file format into the database. Storing the filepath in the database and the hdf5 file to disk works but seems a bit clunky |
The way I do this for PyTorch models is to just return a Python I have plans to provide more flexibility for file storage in the future anyway. An option is to explicitly expose the project's data path to the model, so that it can indeed store the model file under a logical path and just return the directory to the database. |
Thanks - I'll have a look at the byte array approach - I agree keeping all parameters in the database would be a much nicer solution |
Any thoughts on the best approach to integrating a tensorflow model? specifically I mean how to save model weights and architectures to the database. Right now I've saved a filepath in the stateDict then I load and save weights to/from that file. It doesn't seem possible to save the model directly to the db as done for pytorch
The text was updated successfully, but these errors were encountered: