import torch.nn as nn
self.convs = nn.ModuleList([
nn.Conv2d(in_channels = 1,
out_channels = n_filters,
kernel_size = (fs, embedding_dim))
for fs in filter_sizes
])
Yes, you should have something of the shape (batch_size, linear_in)
, and after the linear layer you will have something of the shape (batch_size, linear_out)
, where your linear layer should be declared like nn.Linear(linear_in, linear_out)
.
1 Like