fiftyone.zoo.models.torch

module documentation

(source)

FiftyOne Zoo models provided by torchvision:torchvision.models.

Class	`TorchvisionImageModel`	Wrapper for evaluating a `torchvision:torchvision.models` model on images.
Class	`TorchvisionImageModelConfig`	Configuration for running a `TorchvisionImageModel`.
Variable	`logger`	Undocumented
Function	`_make_load_state_dict_from_url_monkey_patcher`	Monkey patches all instances of `load_state_dict_from_url()` that are reachable from the given `entrypoint` function in the `torchvision:torchvision.models` namespace so that models will be loaded from `model_dir`...

logger = (source) ¶

Undocumented

def _make_load_state_dict_from_url_monkey_patcher(entrypoint, model_dir): (source) ¶

Monkey patches all instances of load_state_dict_from_url() that are reachable from the given entrypoint function in the torchvision:torchvision.models namespace so that models will be loaded from model_dir and not from the Torch Hub cache directory.