Module autodep::torch

source ·
Expand description

Code for loading and running (trained) PyTorch models

Structs

  • A base 64 image
  • A class prediction outputted by a classifier model
  • An in-memory representation of an image (not base 64). Can be the input or output of a model
  • The input to this module’s ML engine – a request for inference
  • Load and run a TorchScript file

Enums

Type Definitions