Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
Can not save model using model.save following multi_gpu_model #8446
Please make sure that the boxes below are checked before you submit your issue. If your issue is an implementation question, please ask your question on StackOverflow or join the Keras Slack channel and ask there instead of filing a GitHub issue.
This is a short and simple issue. Following upgrading to Keras 2.0.9 I have been using the multi_gpu_model utility but I can seem to save my models or best weights using model.save.
The error I get is
I suspect there is some problem gaining access to the model object. Is there a work around this issue?
Yes - thats exactly what I am doing - I'm building the model normally, then calling the multi gpu model method function on it and after training, trying to save it with model.save. Does this need to be converted back to a non-parallelized version of the model? If so, how is this achieved?
That's fantastic, and straightforward advice. Best Simon SLFWalsh MD MRCP FFRRCSI Consultant Radiologist Kings College Hospital Foundation Trust…
On 10 Nov 2017, at 19:09, François Chollet ***@***.***> wrote: For now we recommend saving the original (template) model instead of the parallel model. I.e. call save on the model you passed to multi_gpu_model, not the model returned by it. Both models share the same weights. — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or mute the thread.