def forward()

in model.py [0:0]


    def forward(self, x):
        if self.use_cuda:
            x = x.cuda()

        # embed and transpose for the conv1d layer
        x = self.embed(x)
        x = x.transpose(2,1)

        # feed through conv/prelu layers
        for l in self.local_layers:
            x = l(x)

        # get outputs for model as logits
        y = self.out_layer(x)

        return y