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

Loading different model formats #5

Open
Kali-Zoidberg opened this issue Nov 13, 2023 · 1 comment
Open

Loading different model formats #5

Kali-Zoidberg opened this issue Nov 13, 2023 · 1 comment

Comments

@Kali-Zoidberg
Copy link

I'm not too familiar with the model loading process but is it possible to use this library to load in different model formats (like h5) other than the provided .model extension?

P.S. amazing work on this project. It's been an incredibly useful tool for my organization.

@yuichiis
Copy link
Member

Thank you for finding this project.

As you mentioned, the only model save format is currently ".model".

The h5 you are referring to is probably the model save format used by TensorFlow.

However, this save format only writes out the internal model structure of TensorFlow in a binary file format called HDF5.

Therefore, it cannot be read by machine learning frameworks other than TensorFlow.

For example, TensorFlow models cannot be read by PyTorch. Our framework cannot do it either.

There are projects such as NNEF and ONNX to solve this problem, but I think they have not yet reached a practical state.

It is very difficult to reuse models saved between different frameworks.

I hope our project will be helpful to you.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants