dataeval.utils.torch.models.ResNet18¶

class dataeval.utils.torch.models.ResNet18(embedding_size=128)¶

A wrapper class for the torchvision.models.resnet18 model

Note

This class is provided for the use of DataEval documentation and excludes many features of the torchvision implementation.

Warning

This class has been thoroughly tested for the purposes of DataEval’s documentation but not for operational use. Please use with caution if deploying this class or subclasses.

Parameters:¶

embedding_size : int¶

static transforms()¶

(Returns) the default ResNet18 IMAGENET1K_V1 transforms

Return type:¶

Any