search

AdaptiveMotorControlLab / CEBRA

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

Add device as arg to TensorDataset #170

Closed stes closed 3 months ago

stes commented 3 months ago

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