saving tensorflow models about aerial_wildlife_detection HOT 4 OPEN

The way I do this for PyTorch models is to just return a Python dict with model parameters inside. What you could do is to instead return a byte array of the HDF5 file at the end of the train function (perhaps like described here).
I know this is just as clunky, but at least it would keep the model parameters completely in the database and not just the reference.

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.

from aerial_wildlife_detection.

Hi,

Good question; I remember saving and loading model states to be a bit of a hassle in TensorFlow.
I found the functions model.get_weights() and model.set_weights(), although they are primarily provided for the Keras wrapper: https://www.tensorflow.org/guide/keras/save_and_serialize#weights-only_saving
For TensorFlow itself the functions tf.all_variables() or tf.trainable_variables() might work instead: google/prettytensor#6 (comment)

from aerial_wildlife_detection.

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

from aerial_wildlife_detection.

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

from aerial_wildlife_detection.