search

AdaptiveMotorControlLab / CEBRA

Learnable latent embeddings for joint behavioral and neural analysis - Official implementation of CEBRA
https://cebra.ai
Other
884 stars 72 forks source link

saving and loading models (sklearn API) #33

Closed gonlairo closed 12 months ago

gonlairo commented 1 year ago

Improve model saving in the sklearn API to ensure future compatibility and safety. Currently, we save the entire object using torch.save, which may cause issues with different CEBRA versions. This pull request (PR) tackles this problem by suggesting a solution that involves saving only the model weights and all the arguments used, allowing for complete reconstruction of the CEBRA object at loading time.

Fix https://github.com/AdaptiveMotorControlLab/CEBRA-dev/pull/643

MMathisLab commented 1 year ago

hey @gonlairo could you fix merge conflicts here? thanks!

gonlairo commented 1 year ago

hey @gonlairo could you fix merge conflicts here? thanks!

@MMathisLab we are still developing on dev repo, once its final we'll move it to public and I will fix merge conflifcts.

MMathisLab commented 1 year ago

bump @gonlairo :D

stes commented 12 months ago

closed in favor of the more recent #69