6acdbcfffa5674676dc88c94af3e555f3fa64d17,sru/sru_functional.py,tSRUCell,__init__,#tSRUCell#,617

Before Change

                self.n_proj,
                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
        ))
        self.reset_parameters()

After Change

            ))
        else:
            self.weight_proj = nn.Parameter(torch.Tensor(n_in, self.n_proj))
            self.weight = nn.Parameter(torch.Tensor(
                self.n_proj,
                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)

Instances