9ea14875825b2ab63a43ad2f0f5159d99df86d02,model.py,Lookahead,__init__,#Lookahead#Any#Any#,108

Before Change

        // should we handle batch_first=True?
        super(Lookahead, self).__init__()
        self.n_features = n_features
        self.weight = Parameter(torch.Tensor(n_features, context + 1))
        assert context > 0
        self.context = context
        self.register_parameter("bias", None)
        self.init_parameters()

After Change

        assert context > 0
        self.context = context
        self.n_features = n_features
        self.pad = (0, self.context - 1)
        self.conv = nn.Conv1d(self.n_features, self.n_features, kernel_size=self.context, stride=1,
                              groups=self.n_features, padding=0, bias=None)

    def forward(self, x):

Instances