You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Improvement Description
It could be nice to had support for models that can be loaded on the GPU.
There are a number of libraries such as Pytorch / cuML / skorch that make this pretty easy to implement.
Current Behavior
All of the existing methods are CPU bound, which can become a bottleneck when performing many predictions.
Proposed Behavior
Many of the classifiers in this package already have GPU implementations in skorch / cuML. It would just be a matter of adding the appropriate dependencies / checks / flags to make sure that everything can run.
I wonder if that's really needed. qiime2 is limited in that regard, and introducing pytorch or cuML as an additional huge dependency for the package defies the purpose.
Users unfamiliar with ML will use it on a low n of samples, advanced ML practitioners will start with scikit-learn or pytorch directly.
Improvement Description
It could be nice to had support for models that can be loaded on the GPU.
There are a number of libraries such as Pytorch / cuML / skorch that make this pretty easy to implement.
Current Behavior
All of the existing methods are CPU bound, which can become a bottleneck when performing many predictions.
Proposed Behavior
Many of the classifiers in this package already have GPU implementations in skorch / cuML. It would just be a matter of adding the appropriate dependencies / checks / flags to make sure that everything can run.
References
The text was updated successfully, but these errors were encountered: