search

hereismari / tensorflow-maml

TensorFlow 2.0 implementation of MAML.
Apache License 2.0
83 stars 20 forks source link

Save weight #1

Closed z430 closed 5 years ago

z430 commented 5 years ago

Thanks for sharing your code. I have a question, how to save the MAML weight? I tried save the MAML object using pickle but didn't work

hereismari commented 5 years ago

Hey @z430,

Can you try model.save?

This should work out of the box.

z430 commented 5 years ago

yeah, model.save_weights works