LogME icon indicating copy to clipboard operation
LogME copied to clipboard

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

Open Kyfafyd opened this issue 3 years ago • 3 comments

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 avatar Sep 13 '22 08:09 Kyfafyd

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

youkaichao avatar Sep 13 '22 13:09 youkaichao

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

Kyfafyd avatar Sep 14 '22 03:09 Kyfafyd

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

youkaichao avatar Sep 14 '22 15:09 youkaichao