models icon indicating copy to clipboard operation
models copied to clipboard
verified publisher icon NVIDIA-Merlin

[Task] Generalize TopK to all model types doing multiclass classification in Merlin Models

Open EvenOldridge opened this issue 2 years ago • 0 comments

add a topK operation on top of softmax layer.

EvenOldridge avatar Apr 05 '23 16:04 EvenOldridge