ExTorch.JIT.Model (extorch v0.4.0)

Represents a loaded TorchScript model.

A %ExTorch.JIT.Model{} struct wraps a reference to a torch::jit::script::Module loaded from a .pt file. The model can be used for inference via ExTorch.JIT.forward/2.

Summary

Types

t()

Types

t()

@type t() :: %ExTorch.JIT.Model{
  device: ExTorch.Device.device(),
  reference: reference(),
  resource: any()
}