def apply_init()

in archs/models.py [0:0]

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

        def apply_init(m):
            if isinstance(m, nn.Linear):
                nn.init.normal_(m.weight, 0, 0.1)
                if m.bias is not None:
                    nn.init.constant_(m.bias, 0)
            elif isinstance(m, nn.BatchNorm1d):
                nn.init.constant_(m.weight, 1)
                nn.init.constant_(m.bias, 0)