Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Batched inference CEBRA & padding at the Solver level #168

Open
wants to merge 45 commits into
base: main
Choose a base branch
from
Open

Batched inference CEBRA & padding at the Solver level #168

wants to merge 45 commits into from

Commits on Aug 23, 2024

  1. first proposal for batching in tranform method

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    283de06 View commit details
    Browse the repository at this point in the history

  2. first running version of padding with batched inference

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    202e379 View commit details
    Browse the repository at this point in the history

  3. start tests

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    1f1989d View commit details
    Browse the repository at this point in the history

  4. add pad_before_transform to fit function and add support for convolut… 

    …ional models in _transform
    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    8665660 View commit details
    Browse the repository at this point in the history

  5. remove print statements

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    8d5b114 View commit details
    Browse the repository at this point in the history

  6. first passing test

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    32c5ecd View commit details
    Browse the repository at this point in the history

  7. add support for hybrid models

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    9928f63 View commit details
    Browse the repository at this point in the history

  8. rewrite transform in sklearn API

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    be5630a View commit details
    Browse the repository at this point in the history

  9. baseline version of a torch.Datset

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    1300b20 View commit details
    Browse the repository at this point in the history

  10. move batching logic outside solver

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    bc6af24 View commit details
    Browse the repository at this point in the history

  11. move functionality to base file in solver and separate in functions

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    ec377b9 View commit details
    Browse the repository at this point in the history

  12. add test_select_model for single session

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    6f9ca98 View commit details
    Browse the repository at this point in the history

  13. add checks and test for _process_batch

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    fbe7eb4 View commit details
    Browse the repository at this point in the history

  14. add test_select_model for multisession

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    463b0f8 View commit details
    Browse the repository at this point in the history

  15. make self.num_sessions compatible with single session training

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    5219171 View commit details
    Browse the repository at this point in the history

  16. improve test_batched_transform_singlesession

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    f9bd1a6 View commit details
    Browse the repository at this point in the history

  17. make it work with small batches

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    e23a7ef View commit details
    Browse the repository at this point in the history

  18. make test with multisession work

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    19c3f87 View commit details
    Browse the repository at this point in the history

  19. change to torch padding

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    87bebac View commit details
    Browse the repository at this point in the history

  20. add argument to sklearn api

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    f0303e0 View commit details
    Browse the repository at this point in the history

  21. add torch padding to _transform

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    8c8be85 View commit details
    Browse the repository at this point in the history

  22. convert to torch if numpy array as inputs

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    59df402 View commit details
    Browse the repository at this point in the history

  23. add distinction between pad with data and pad with zeros and modify t… 

    …est accordingly
    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    1aadc8b View commit details
    Browse the repository at this point in the history

  24. differentiate between data padding and zero padding

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    bc8ee25 View commit details
    Browse the repository at this point in the history

  25. remove float16

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    5e7a14c View commit details
    Browse the repository at this point in the history

  26. change argument position

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    928d882 View commit details
    Browse the repository at this point in the history

  27. clean test

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    07bac1c View commit details
    Browse the repository at this point in the history

  28. clean test

    @gonlairo @stes
    gonlairo authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    0823b54 View commit details
    Browse the repository at this point in the history

  29. Fix warning

    @CeliaBenquet @stes
    CeliaBenquet authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    9fe3af3 View commit details
    Browse the repository at this point in the history

  30. Improve modularity remove duplicate code and todos

    @CeliaBenquet @stes
    CeliaBenquet authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    b417a23 View commit details
    Browse the repository at this point in the history

  31. Add tests to solver

    @CeliaBenquet @stes
    CeliaBenquet authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    83c1669 View commit details
    Browse the repository at this point in the history

  32. Remove unused import in solver/utils

    @CeliaBenquet @stes
    CeliaBenquet authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    9c46eb9 View commit details
    Browse the repository at this point in the history

  33. Fix test plot

    @CeliaBenquet @stes
    CeliaBenquet authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    c845ec3 View commit details
    Browse the repository at this point in the history

  34. Add some coverage

    @CeliaBenquet @stes
    CeliaBenquet authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    9db3e37 View commit details
    Browse the repository at this point in the history

  35. Fix save/load

    @CeliaBenquet @stes
    CeliaBenquet authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    8e5f933 View commit details
    Browse the repository at this point in the history

  36. Remove duplicate configure_for in multi dataset

    @CeliaBenquet @stes
    CeliaBenquet authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    d08e400 View commit details
    Browse the repository at this point in the history

  37. Make save/load cleaner

    @CeliaBenquet @stes
    CeliaBenquet authored and stes committed Aug 23, 2024
    Configuration menu
    Copy the full SHA
    0c693dd View commit details
    Browse the repository at this point in the history

Commits on Sep 18, 2024

  1. Merge branch 'main' into batched-inference-and-padding

    @CeliaBenquet
    CeliaBenquet committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    ae056b2 View commit details
    Browse the repository at this point in the history

  2. Fix codespell errors

    @CeliaBenquet
    CeliaBenquet committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    794867b View commit details
    Browse the repository at this point in the history

  3. Fix docs compilation errors

    @CeliaBenquet
    CeliaBenquet committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    0bb6549 View commit details
    Browse the repository at this point in the history

  4. Fix formatting

    @CeliaBenquet
    CeliaBenquet committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    04a102f View commit details
    Browse the repository at this point in the history

  5. Fix extra docs errors

    @CeliaBenquet
    CeliaBenquet committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    7aab282 View commit details
    Browse the repository at this point in the history

  6. Fix offset in docs

    @CeliaBenquet
    CeliaBenquet committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    ffa66eb View commit details
    Browse the repository at this point in the history

  7. Remove attribute ref

    @CeliaBenquet
    CeliaBenquet committed Sep 18, 2024
    Configuration menu
    Copy the full SHA
    7f58607 View commit details
    Browse the repository at this point in the history

Commits on Sep 19, 2024

  1. Add review updates

    @CeliaBenquet
    CeliaBenquet committed Sep 19, 2024
    Configuration menu
    Copy the full SHA
    c2544c7 View commit details
    Browse the repository at this point in the history