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¶