rexnet1_0x¶

holocron.models.rexnet1_0x(pretrained: bool = False, checkpoint: Checkpoint | None = None, progress: bool = True, **kwargs: Any) → ReXNet[source]¶

ReXNet-1.0x from “ReXNet: Diminishing Representational Bottleneck on Convolutional Neural Network”

Parameters:

pretrained (bool) – If True, returns a model pre-trained on ImageNette
checkpoint – If specified, the model’s parameters will be set to the checkpoint’s values
progress (bool) – If True, displays a progress bar of the download to stderr
kwargs – keyword args of _rexnet

Returns:

classification model

Return type:

torch.nn.Module

class holocron.models.ReXNet1_0x_Checkpoint(value)[source]¶

The model builder above accepts the following values as the checkpoint parameter. ReXNet1_0x_Checkpoint.DEFAULT is equivalent to ReXNet1_0x_Checkpoint.IMAGENET1K

ReXNet1_0x_Checkpoint.IMAGENET1K:

Also available as ReXNet1_0x_Checkpoint.DEFAULT.

dataset	imagenet-1k
top1-accuracy	0.7786
top5-accuracy	0.9387
url	link
sha256	ab7b973341a59832
size	13.7MB
num_params	4.8M
categories	tench, goldfish, great white shark, … (997 omitted)
input_shape	(3, 224, 224)
mean	(0.485, 0.456, 0.406)
std	(0.229, 0.224, 0.225)
interpolation	InterpolationMode.BILINEAR
commit	None

ReXNet1_0x_Checkpoint.IMAGENETTE:

dataset	imagenette
top1-accuracy	0.9439
top5-accuracy	0.9962
url	link
sha256	7c19fd53a5433927
size	13.7MB
num_params	3.5M
categories	tench, English springer, cassette player, … (7 omitted)
input_shape	(3, 224, 224)
mean	(0.485, 0.456, 0.406)
std	(0.229, 0.224, 0.225)
interpolation	InterpolationMode.BILINEAR
commit	d4a5999