lenskit.util.torch#
PyTorch utility functions.
Functions
|
Function decorator that puts PyTorch in inference mode. |
- lenskit.util.torch.inference_mode(func)#
Function decorator that puts PyTorch in inference mode.
PyTorch utility functions.
Functions
|
Function decorator that puts PyTorch in inference mode. |
Function decorator that puts PyTorch in inference mode.