Closed blondered closed 2 days ago
use_gpu argument to ImplicitRanker.rank method that calls implicit.gpu.KnnQuery().topk instead of implicit.cpu.topk.topk when set to True.
use_gpu
ImplicitRanker.rank
implicit.gpu.KnnQuery().topk
implicit.cpu.topk.topk
Allows for gpu inference for a range of models
Described at https://github.com/MobileTeleSystems/RecTools/issues/199
Feature Description
use_gpuargument toImplicitRanker.rankmethod that callsimplicit.gpu.KnnQuery().topkinstead ofimplicit.cpu.topk.topkwhen set to True.Why this feature?
Allows for gpu inference for a range of models
Additional context
Described at https://github.com/MobileTeleSystems/RecTools/issues/199