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.

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

Update vision mamba backbone to latest updates #401

Merged
merged 10 commits into from
Nov 13, 2024

Conversation

anwai98
Copy link
Collaborator

@anwai98 anwai98 commented Nov 3, 2024

This PR focuses on two things:

  • Adapt ViM-UNet's vim encoder to the latest legacy updates (the latest changes update some implementation details which in principle speed up the training time)

I decided to leave out the part of refactoring the installation step (in principle, I would like to move all installations in a config file which takes care of it all)

@anwai98 anwai98 self-assigned this Nov 3, 2024
@anwai98 anwai98 marked this pull request as ready for review November 11, 2024 13:25
@anwai98
Copy link
Collaborator Author

anwai98 commented Nov 11, 2024

I did a sanity check training experiment on LIVECell, and I can reproduce numbers from our paper. This is now GTG from my side. Thanks!

Copy link
Owner

@constantinpape constantinpape left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I didn't check the changes in much detail, but looks good overall.

@constantinpape constantinpape merged commit 00c7218 into constantinpape:main Nov 13, 2024
2 checks passed
@anwai98 anwai98 deleted the u-vim branch November 13, 2024 13:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants