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

Model 2 subbatch_size not working #64

Open
smilenaderi opened this issue Apr 12, 2023 · 3 comments
Open

Model 2 subbatch_size not working #64

smilenaderi opened this issue Apr 12, 2023 · 3 comments

Comments

@smilenaderi
Copy link

Hi I'm trying to inference a sequence with length 548 on an A10 GPU with model 2
I tried subbatch size equal to 1 but GRAM is low and it can not be runned.
@wewewexiao2008
Copy link

wewewexiao2008 commented Apr 30, 2023 •
edited
Loading

@RuiWang1998 Same problem. --subbatch_size seems not working on model 2, but fine with model 1.

@KYQiu21
Copy link

KYQiu21 commented Jul 30, 2023 •
edited
Loading

Have you solved this? A length of 300AA proteins with subbatch_size 128 reported the memory error on an A100-40G.

@smilenaderi
Copy link
Author

No, I'm just working with model 1

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@wewewexiao2008 @KYQiu21 @smilenaderi