ResNetInput¶
-
class
torchelie.models.ResNetInput(in_channels: int = 3, out_channels: int = 64)¶ -
forward(x: torch.Tensor) → torch.Tensor¶
-
set_input_specs(input_size: int, in_channels=3) → torchelie.models.resnet.ResNetInput¶
-
set_stride(stride: int) → torchelie.models.resnet.ResNetInput¶
-
conv: torchelie.nn.conv.ConvBlock¶
-
pool: Optional[torch.nn.modules.module.Module]¶
-