def __call__()

in jcm/models/wideresnet_noise_conditional.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

    def __call__(self, x, temb=None, train=True):
        for i in range(self.blocks_per_group):
            x = WideResnetBlock(
                self.channels,
                self.strides if i == 0 else (1, 1),
                activate_before_residual=self.activate_before_residual and not i,
            )(x, temb, train)
        return x