dacapo.experiments.architectures.dummy_architecture
Module Contents
Classes
A class used to represent a dummy architecture layer for a 3D CNN. |
- class dacapo.experiments.architectures.dummy_architecture.DummyArchitecture(architecture_config)
A class used to represent a dummy architecture layer for a 3D CNN.
- channels_in
An integer representing the number of input channels.
- channels_out
An integer representing the number of output channels.
- conv
A 3D convolution object.
- input_shape
A coordinate object representing the shape of the input.
- forward(x)
Performs the forward pass of the network.
- property input_shape
Returns the input shape for this architecture.
- Returns:
Input shape of the architecture.
- Return type:
Coordinate
- property num_in_channels
Returns the number of input channels for this architecture.
- Returns:
Number of input channels.
- Return type:
int
- property num_out_channels
Returns the number of output channels for this architecture.
- Returns:
Number of output channels.
- Return type:
int
- forward(x)
Perform the forward pass of the network.
- Parameters:
x – Input tensor.
- Returns:
Output tensor after the forward pass.
- Return type:
Tensor