Edit on GitHub

Torch

ModelType and ModelIO implementations for torch.nn.Module ImportHook for importing files saved with torch.save DataType, Reader and Writer implementations for torch.Tensor

Description

TODO

Requirements

$ pip install mlem[torch]
# or
$ pip install torch
Content

šŸ› Found an issue? Let us know! Or fix it:

Edit on GitHub

ā“ Have a question? Join our chat, we will help you:

Discord Chat