b499c2fd5b0907290d0778ec8c76144134f3d776,hypergan/configurable_component.py,ConfigurableComponent,activation_to_module,#ConfigurableComponent#Any#,188

Before Change

        if name == "relu":
            return nn.ReLU()
        if name == "tanh":
            return nn.Tanh()

    def count_number_trainable_params(self):
        """
        Counts the number of trainable variables.

After Change

        self.named_layers[name]     = net

    def activation_to_module(self, name):
        return {
            "celu": nn.CELU,
            "gelu": nn.GELU,
            "lrelu": nn.LeakyReLU,
            "prelu": nn.PReLU,
            "relu": nn.ReLU,
            "relu6": nn.ReLU6,
            "selu": nn.SELU,
            "sigmoid": nn.Sigmoid,
            "softplus": nn.Softplus,
            "softshrink": nn.Softshrink,
            "softsign": nn.Softsign,
            "tanh": nn.Tanh,
            "tanhshrink": nn.Tanhshrink
        }[name]()

    def count_number_trainable_params(self):
        """
        Counts the number of trainable variables.

Instances