class documentation
class TorchModelMixin(object): (source)
Known subclasses: fiftyone.utils.torch.TorchImageModel
Mixin for Model classes that support feeding data for inference
via a torch:torch.utils.data.DataLoader.
Models implementing this mixin must expose via their
Model.transforms property the
torchvision:torchvision.transforms function that will/must be
applied to each input before prediction.