![]() |
Belle II Software development
|
Public Member Functions | |
__init__ (self, output_size, input_size, which_linear, eps=1e-5, momentum=0.1, cross_replica=False, mybn=False, norm_style="bn") | |
Constructor. | |
forward (self, x, y) | |
forward | |
extra_repr (self) | |
extra_repr | |
Public Attributes | |
output_size | |
output size | |
input_size = output_size, input_size | |
output size | |
gain = which_linear(input_size, output_size) | |
Prepare gain and bias layers. | |
bias = which_linear(input_size, output_size) | |
bias | |
eps = eps | |
epsilon to avoid dividing by 0 | |
momentum = momentum | |
Momentum. | |
cross_replica = cross_replica | |
Use cross-replica batchnorm? | |
mybn = mybn | |
Use my batchnorm? | |
str | norm_style = norm_style |
Norm style? | |
bn = myBN(output_size, self.eps, self.momentum) | |
bn | |
Private Attributes | |
__dict__ | |
Class-conditional bn output size is the number of channels, input size is for the linear layers Andy's Note: this class feels messy but I'm not really sure how to clean it up # noqa Suggestions welcome! (By which I mean, refactor this and make a merge request if you want to make this more readable/usable).
__init__ | ( | self, | |
output_size, | |||
input_size, | |||
which_linear, | |||
eps = 1e-5, | |||
momentum = 0.1, | |||
cross_replica = False, | |||
mybn = False, | |||
norm_style = "bn" ) |
Constructor.
Definition at line 526 of file ieagan.py.
extra_repr | ( | self | ) |
forward | ( | self, | |
x, | |||
y ) |
forward
Definition at line 563 of file ieagan.py.
bias = which_linear(input_size, output_size) |
bn = myBN(output_size, self.eps, self.momentum) |
cross_replica = cross_replica |
gain = which_linear(input_size, output_size) |
input_size = output_size, input_size |