Batched inference CEBRA & padding at the `Solver` level#168

Open

CeliaBenquet wants to merge 49 commits intoAdaptiveMotorControlLab:mainfrom CeliaBenquet:batched-inference-and-padding

+1,430-147

Commits on Aug 23, 2024

first proposal for batching in tranform method

gonlairo
authored and
stes
committed
first running version of padding with batched inference

gonlairo
authored and
stes
committed
start tests

gonlairo
authored and
stes
committed
add pad_before_transform to fit function and add support for convolutional models in _transform

gonlairo
authored and
stes
committed
remove print statements

gonlairo
authored and
stes
committed
first passing test

gonlairo
authored and
stes
committed
add support for hybrid models

gonlairo
authored and
stes
committed
rewrite transform in sklearn API

gonlairo
authored and
stes
committed
baseline version of a torch.Datset

gonlairo
authored and
stes
committed
move batching logic outside solver

gonlairo
authored and
stes
committed
move functionality to base file in solver and separate in functions

gonlairo
authored and
stes
committed
add test_select_model for single session

gonlairo
authored and
stes
committed
add checks and test for _process_batch

gonlairo
authored and
stes
committed
add test_select_model for multisession

gonlairo
authored and
stes
committed
make self.num_sessions compatible with single session training

gonlairo
authored and
stes
committed
improve test_batched_transform_singlesession

gonlairo
authored and
stes
committed
make it work with small batches

gonlairo
authored and
stes
committed
make test with multisession work

gonlairo
authored and
stes
committed
change to torch padding

gonlairo
authored and
stes
committed
add argument to sklearn api

gonlairo
authored and
stes
committed
add torch padding to _transform

gonlairo
authored and
stes
committed
convert to torch if numpy array as inputs

gonlairo
authored and
stes
committed
add distinction between pad with data and pad with zeros and modify test accordingly

gonlairo
authored and
stes
committed
differentiate between data padding and zero padding

gonlairo
authored and
stes
committed
remove float16

gonlairo
authored and
stes
committed
change argument position

gonlairo
authored and
stes
committed
clean test

gonlairo
authored and
stes
committed
clean test

gonlairo
authored and
stes
committed
Fix warning

CeliaBenquet
authored and
stes
committed
Improve modularity remove duplicate code and todos

CeliaBenquet
authored and
stes
committed
Add tests to solver

CeliaBenquet
authored and
stes
committed
Remove unused import in solver/utils

CeliaBenquet
authored and
stes
committed
Fix test plot

CeliaBenquet
authored and
stes
committed
Add some coverage

CeliaBenquet
authored and
stes
committed
Fix save/load

CeliaBenquet
authored and
stes
committed
Remove duplicate configure_for in multi dataset

CeliaBenquet
authored and
stes
committed
Make save/load cleaner

CeliaBenquet
authored and
stes
committed

Commits on Sep 19, 2024

Add review updates
CeliaBenquet
committed

Commits on Oct 20, 2024

Commits on Oct 27, 2024