fast-soft-sort icon indicating copy to clipboard operation
fast-soft-sort copied to clipboard

Published 20 hours ago • verified publisher icon google-research

Does it support fp16 training?

Open SkyAndCloud opened this issue 3 years ago • 2 comments

Hi, thank you for the efforts. I want to use fast_soft_sort in fp16 setting. It seems that fast_soft_sort doesn't support fp16 yet? Looking forward to your reply!

SkyAndCloud avatar Aug 16 '21 07:08 SkyAndCloud

Is it for tensorflow, pytorch or JAX?

CC @josipd

mblondel avatar Aug 16 '21 07:08 mblondel

Thank you for the quick reply. This question is for pytorch

SkyAndCloud avatar Aug 17 '21 03:08 SkyAndCloud