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

AutoChunk #160

Open
georgkempf opened this issue Mar 5, 2023 · 2 comments
Open

AutoChunk #160

georgkempf opened this issue Mar 5, 2023 · 2 comments
Labels
enhancement New feature or request

Comments

@georgkempf
Copy link

Hello, I was wondering whether AutoChunk described in https://arxiv.org/pdf/2203.00854v3.pdf is already implemented? Will this avoid manually setting the chunk_size?

@oahzxl
Copy link
Contributor

oahzxl commented Mar 7, 2023

Hi, autochunk was implemented in colossalAI, and it has not been added to this repo. So I'm afraid you still need to set it manually now.

@Gy-Lu Gy-Lu added the enhancement New feature or request label Mar 23, 2023
@ssn1771
Copy link

ssn1771 commented Jan 5, 2024

@oahzxl Hi, I was curious about the inference overhead caused by using AutoChunk for protein prediction models , did it take some time to find the optimal chunk size?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

4 participants