-
Notifications
You must be signed in to change notification settings - Fork 146
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
Add an example of object segmentation (ClipSeg) #801
Conversation
5eb2f35
to
80ca06b
Compare
@regisss |
Add GPU Benchmark: Nvidia A100-80GB, CUDA 12.2, Driver Version: 535.54.03 With BF16 autocast: |
@cfgfung Your result looks good. Could you also port relevant test case for your model and make sure those are passing? |
Actually, above model test cases are optional, it's good to check but not necessary. |
Hi, I have added the test and followed the same format as the PR of SAM. |
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
@cfgfung all looks good. Could you please make sure that you ran "make style" and upload the fix? I still see many style error. |
Hi, There might be some updates in the rules and hence triggered the new errors. |
Co-authored-by: regisss <[email protected]>
Co-authored-by: regisss <[email protected]>
Co-authored-by: regisss <[email protected]>
Co-authored-by: regisss <[email protected]>
Co-authored-by: regisss <[email protected]>
What does this PR do?
The ClipSeg model has been validated using graph mode on Gaudi 2. Adding this as an example for users.
Performance number:
Without graph mode
With graph mode
Fixes # (issue)
N.A.
Before submitting