Saving/Loading with utils.checkpoint

Guillem96 / efficientdet-tf

Custom implementation of EfficientDet https://arxiv.org/abs/1911.09070

GNU General Public License v3.0

96 stars 23 forks source link

Saving/Loading with utils.checkpoint #19

Closed JamesDConley closed 3 years ago

JamesDConley commented 4 years ago

the utils.checkpoint.save saves in .tf format while utils.checkpoint.load expects .h5 format. Is there another way to save/load a model?

Guillem96 commented 4 years ago

I am currently working with checkpointing, but I ve been a bit busy this days. In the following days, I'll try to implement a consistent checkpoint API

JamesDConley commented 4 years ago

Thank you!