fannypack.nn¶
Residual Blocks¶
-
class
fannypack.nn.resblocks.
Base
(activation: str = 'relu', activations_inplace: bool = False)¶ Bases:
torch.nn.modules.module.Module
,abc.ABC
Base class for a generic residual block, with support for
"relu"
,"leaky_relu"
,"selu"
, and"none"
activations.-
forward
(x: torch.Tensor) → torch.Tensor¶ ResBlock forward pass.
-
-
class
fannypack.nn.resblocks.
Conv2d
(channels: int, bottleneck_channels: Optional[int] = None, kernel_size: int = 3, **resblock_base_args)¶ Bases:
fannypack.nn.resblocks.Base
Standard convolutional residual block.
-
class
fannypack.nn.resblocks.
Linear
(units: int, bottleneck_units: Optional[int] = None, **resblock_base_args)¶ Bases:
fannypack.nn.resblocks.Base
Standard linear residual block.