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

Why the LogME is implemented not in torch but in numpy? #18

Open
Kyfafyd opened this issue Sep 13, 2022 · 3 comments
Open

Why the LogME is implemented not in torch but in numpy? #18

Kyfafyd opened this issue Sep 13, 2022 · 3 comments

Comments

@Kyfafyd
Copy link

Kyfafyd commented Sep 13, 2022

RT.
Maybe with CUDA, LogME costs less time? Or if this is the problem setting assuming the transferability evaluation can not access GPU? (but the feature extraction part needs GPU?)

@Kyfafyd Kyfafyd changed the title Why the LogME is implemented in torch but in numpy? Why the LogME is implemented not in torch but in numpy? Sep 13, 2022
@youkaichao
Copy link
Member

You are welcome to open a pull request with CUDA implementation.

@Kyfafyd
Copy link
Author

Kyfafyd commented Sep 14, 2022

Thanks for response!
I would like to further learn if using torch implementation for comparing efficiency is fair?

@youkaichao
Copy link
Member

You can compare torch naive implementation and torch accelerated implementation using our proposed algorithm of fixed point iteration.

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

No branches or pull requests

2 participants