ResNet¶
-
class
torchelie.models.ResNet(arch: List[str], num_classes: int)¶ -
add_se() → torchelie.models.resnet.ResNet¶
-
forward(x: torch.Tensor) → torch.Tensor¶
-
set_input_specs(input_size: int = 224, in_channels: int = 3) → torchelie.models.resnet.ResNet¶
-
to_bottleneck() → torchelie.models.resnet.ResNet¶
-
to_preact() → torchelie.models.resnet.ResNet¶
-
to_preact_bottleneck() → torchelie.models.resnet.ResNet¶
-
to_preact_resnext() → torchelie.models.resnet.ResNet¶
-
to_preact_wide() → torchelie.models.resnet.ResNet¶
-
to_resnext() → torchelie.models.resnet.ResNet¶
-
to_wide() → torchelie.models.resnet.ResNet¶
-
use_standard_input()¶
-
training: bool¶
-