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

Preprocessing on GPU #5

Open
zclawr opened this issue Nov 7, 2024 · 0 comments
Open

Preprocessing on GPU #5

zclawr opened this issue Nov 7, 2024 · 0 comments
Assignees

Comments

@zclawr
Copy link
Contributor

zclawr commented Nov 7, 2024

Try converting np arrays to PyTorch tensors and running preprocessing on these, then convert back to np array for passing into the model. Ideally get some number comparisons in terms of performance between the CPU preprocessing and GPU preprocessing.

@anavani27 anavani27 self-assigned this Nov 15, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants