fc850582ced2b873507493a7a6eafd7bcc9a24e2,sru/sru_functional.py,tSRUCell,__init__,#tSRUCell#Any#Any#Any#Any#Any#Any#Any#,610

Before Change

                n_out*4 if bidirectional else n_out*2
            ))
        self.weight_c = nn.Parameter(torch.Tensor(
            n_out*2 if bidirectional else n_out
        ))
        self.bias = nn.Parameter(torch.Tensor(
            n_out*2 if bidirectional else n_out
        ))

After Change

                out_size*2
            ))
        self.weight_c = nn.Parameter(torch.Tensor(out_size))
        self.bias = nn.Parameter(torch.Tensor(out_size))
        if layer_norm:
            self.layer_norm = nn.LayerNorm(self.n_in)
        else:
            self.layer_norm = None

Instances