simulation.utils.machine_learning.models.resnet_block module¶

Summary¶

Classes:

ResnetBlock

Define a Resnet block.

Reference¶

class ResnetBlock(dim: int, padding_type: str, norm_layer: Type[torch.nn.modules.module.Module], use_dropout: bool, use_bias: bool, n_conv_layers: int = 2, dilations: Optional[List[int]] = None)[source]¶

Bases: torch.nn.modules.module.Module

Define a Resnet block.

forward(x: torch.Tensor) → torch.Tensor[source]¶

Standard forward with skip connection.

Parameters: x – Input tensor

training: bool¶