ABOD (potentially also others) fails when the (default) device is cuda because the indices generated by basic_operators.bottomk are always on cpu:
|
return tk[0].cpu(), tk[1].cpu() |
Similarly for basic_operators.topk:
|
return tk[0].cpu(), tk[1].cpu() |
Both functions do not respect the device kwarg.
ABOD (potentially also others) fails when the (default) device is cuda because the indices generated by
basic_operators.bottomkare always on cpu:pytod/pytod/models/basic_operators.py
Line 84 in ec43433
Similarly for
basic_operators.topk:pytod/pytod/models/basic_operators.py
Line 75 in ec43433
Both functions do not respect the device kwarg.