How to convert the auto-encoder h5 file to pytorch format?

Question

How to convert the auto-encoder h5 file to pytorch format?

Closed this issue 5 years ago · 1 comments

I am reproduce the AutoZoom code to pytorch version, but the biggest challenge is the h5 file that is pre-trained by the repo owner. How to train the auto-encoder and which dataset do you use?
Or can I convert the h5 file to the model file that pytorch model can load?

Answer 1 · 2020-01-27T15:53:50.000Z

Hello,

Thank you for being interested in our work.

We only have the tensorflow support for now. You are welcome to contribute the pytorch version.