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