Model loading error

[MingxinDing]

Is there an existing issue for this?

• I have searched the existing issues

Bug description

New commit failed to load model files.
cebra_error.md

Operating System

Windows 11

CEBRA version

0.4.0

Device type

gpu

Steps To Reproduce

No response

Relevant log output

No response

Anything else?

No response

Code of Conduct

• I agree to follow this project's Code of Conduct

[stes]

Thanks for flagging!

To understand the issue, you saved the model using a previous cebra version, and try to load it with the latest commit?

I'll attempt to repro the issue on my end.

[MingxinDing]

Yes I did originally. Then I saved models using the latest commit, same error with loading those as well.
Thanks very much!

[MMathisLab]

Which version of pytorch do you have installed? Thanks!

[MingxinDing]

2.5.1+cu121

[stes]

The error most likely occurred in #205 where the general logic behind the CEBRA model changed a bit. We had to adapt this due to some API changes in sklearn. I will see if we can ensure compatibility of model loading and keep you updated here.

In general, thanks for catching this, important issue to fix to keep the loading function backward compatible!