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
- The output of a model’s inference
- The type of inference to compute
- Input data that inference can be computed on