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

Save the model #4

Open
lanyiyun opened this issue May 17, 2018 · 1 comment
Open

Save the model #4

lanyiyun opened this issue May 17, 2018 · 1 comment

Comments

@lanyiyun
Copy link

Hi,

I saw you have some pretrained model saved in h5. I wonder how do you save and load the SpectralNet model given that the last layer is a customized Lambda layer? The save() or to_json() or to_yaml() would be invalid. Thank you in advance!

@lihenryhfl
Copy link
Collaborator

Late response, but the h5 pretrained models are only for the autoencoder. To level the playing field, we use the same autoencoder as several of the papers that we compare ourselves to. We have not tried saving the SpectralNet model itself, though a search on the web suggests that there is a chance it is possible.

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