Edit on GitHub

torch

class TorchModel

MlemABC parent type: model_type

MlemABC type: torch

:class:`.ModelType` implementation for PyTorch models

Fields:

  • io: ModelIO = TorchModelIO() - TorchModelIO

class TorchModelIO

MlemABC parent type: model_io

MlemABC type: torch_io

IO for PyTorch models

Fields:

  • is_jit: bool = False - Is model jit compiled
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