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

Concatenate last batches for batched inference #200

Open CeliaBenquet opened 1 week ago

CeliaBenquet commented 1 week ago

fix #199.

To still transform the extra samples that are not contained in a full batch while not getting an error regarding the next to last batch batch_end_idx above the length of the input, I propose to concatenate the penultimate batch (so the last complete batch) with the last batch (the incomplete batch from 1 to batch_size).