Hi,
according to Understanding Convolution 1D output and Input my input shape [6, 3, 1] corresponds to [batch_size, in_channel, len]. In addition, out_channel defines the number of the kernels.
As a consequence, MyNet should become:
MyNet = nn.Sequential(
nn.Conv1d(in_channels = 3, out_channels = 64, kernel_size = 2),
nn.MaxPool1d(2),
nn.Flatten(start_dim=1),
nn.Linear(64,50),
nn.ReLU(),
nn.Linear(50, 1)
)
Is it correct?
It doesn’t work, I get the following error: Calculated padded input size per channel: (1). Kernel size: (2). Kernel size can't be greater than actual input size
Please, let me know what I am doing wrong!