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

Use PZMM to Save Tensorflow Model #108

Open
SophiaRowland opened this issue Oct 1, 2021 · 0 comments
Open

Use PZMM to Save Tensorflow Model #108

SophiaRowland opened this issue Oct 1, 2021 · 0 comments
Labels
enhancement New feature or request

Comments

@SophiaRowland
Copy link
Collaborator

I was working with a customer who was trying to use the pzmm.PickleModel.pickleTrainedModel() function on a TensorFlow model, but they were getting a 'TypeError: can't pickle weakref objects'. My ideal solution would be to for the PickleTrainedModel() function to have the ability to ingest Tensorflow models or have a similar function available for Tensorflow models. For now, I think the work-around would be to use Tensorflow's save() function to save the model, make that model available to Model Manager, and then edit the score code to use Tensorflow's load_model() function instead of un-pickling the model, but I am open to other work-around ideas.

@SophiaRowland SophiaRowland added the enhancement New feature or request label Oct 1, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant