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

[BUG] TFT model load failed from saved model. #1946

Closed
heury opened this issue Aug 10, 2023 · 1 comment · Fixed by #1952
Closed

[BUG] TFT model load failed from saved model. #1946

heury opened this issue Aug 10, 2023 · 1 comment · Fixed by #1952
Labels
bug Something isn't working

Comments

@heury
Copy link

heury commented Aug 10, 2023

Describe the bug

I get following error when I try to load TFT model from file.

ValueError: The values of the hyper parameter categorical_embedding_sizes should be identical between the instantiated model (None) and the loaded checkpoint ({}). Please adjust the model accordingly.

To Reproduce
I just save trained model and load it from file.
And I get following error when I try to load model from file.

ValueError: The values of the hyper parameter categorical_embedding_sizes should be identical between the instantiated model (None) and the loaded checkpoint ({}). Please adjust the model accordingly.

This error is gone when I add following parameter to the model.

categorical_embedding_sizes={}

Expected behavior

System (please complete the following information):

  • Python version: 3.10
  • darts version: 0.25

Additional context
Add any other context about the problem here.

@heury heury added bug Something isn't working triage Issue waiting for triaging labels Aug 10, 2023
@madtoinou madtoinou removed the triage Issue waiting for triaging label Aug 10, 2023
@dennisbader
Copy link
Collaborator

Hi @heury and thanks pointing this out. It's indeed a bug and we'll fix it soon.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
Status: Released
Development

Successfully merging a pull request may close this issue.

3 participants