![]() |
Belle II Software development
|
Class for the ResNet generator model. More...
Public Member Functions | |
__init__ (self) | |
Constructor to create a new model instance. | |
forward (self, z) | |
Function to perform a forward pass. | |
Public Attributes | |
fc = nn.Linear(96, 49152) | |
Fully-connected layer. | |
blocks | |
Sequence of residual block layers. | |
norm = nn.BatchNorm2d(16) | |
Batch normalization layer. | |
conv = nn.Conv2d(16, 1, 3, 1, 1) | |
Convolutional layer. | |
Class for the ResNet generator model.
ResNet generator model.
__init__ | ( | self | ) |
Constructor to create a new model instance.
Definition at line 91 of file resnet.py.
forward | ( | self, | |
z ) |
Function to perform a forward pass.
Compute the model output for a given input.
Definition at line 117 of file resnet.py.
blocks |
Sequence of residual block layers.