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

[Enhancement]: Support GPU in data collection #1158

Closed
1 task done
reiase opened this issue Apr 27, 2022 · 3 comments
Closed
1 task done

[Enhancement]: Support GPU in data collection #1158

reiase opened this issue Apr 27, 2022 · 3 comments
Labels
kind/enhancement Issues or changes related to enhancement stale

Comments

@reiase
Copy link
Contributor

reiase commented Apr 27, 2022

Is there an existing issue for this?

  • I have searched the existing issues

What would you like to be added?

Support GPU in data collection, for example:

towhee.range(10) \
    .CPU.image_decode() \
    .GPU(0).image_embedding.vgg() \
    .to_list()

Why is this needed?

No response

Anything else?

No response

@reiase reiase added the kind/enhancement Issues or changes related to enhancement label Apr 27, 2022
@filip-halt
Copy link
Contributor

filip-halt commented Apr 27, 2022

Is callholder able to look indefinitely deep in the chain. In this example im assuming getattr will be called on GPU.image_embedding.vgu?

@reiase
Copy link
Contributor Author

reiase commented Apr 28, 2022

It can't, but we can modify the call holder if necessary.

@stale
Copy link

stale bot commented May 28, 2022

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions. Rotten issues close after 30d of inactivity. Close the stale issues and pull requests after 7 days of inactivity. Reopen the issue with /reopen.

@stale stale bot added the stale label May 28, 2022
@stale stale bot closed this as completed Jun 4, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
kind/enhancement Issues or changes related to enhancement stale
Projects
None yet
Development

No branches or pull requests

2 participants