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

Can CEBRA use 2d or 3d position as label? #63

Closed cnwangy closed 1 year ago

cnwangy commented 1 year ago

Can CEBRA use 2d or 3d position as label? If I change the label to 2d delta, an error will be thrown

RuntimeError: The size of tensor a (2) must match the size of tensor b (4) at non-singleton dimension 1

MMathisLab commented 1 year ago

Hi @cnwangy thanks for your question. Could you provide us with example code and failed output, so we can better understand your query, thanks!

stes commented 1 year ago

Hi @cnwangy , for now I would recommend to switch to the time_delta distribution, then you can use higher dimensional outputs.

We might add support for higher dimensional labels when using the delta distribution in the future, right now this is not supported.

cnwangy commented 1 year ago

Thanks a lot! I will give a try.