Hi,
I am referring to your code and have some questions regarding forward function.
I am trying to apply VGG11 on CIFAR-10 dataset.
I applied the Average pooling layer after 2 Convolutional layers.
But in forward function, I am giving the spike of the first conv layer to the second conv layer and then applying avg pool layer to the second spike of conv layer.
In this case, my code is not giving any output and error.
I don't know, what's wrong with this forward function and couldn't fix it.
Can you please tell me what should I do?
Your help will be appreciated.
Below is my code:
class SCNN(nn.Module):
def init(self):
super(SCNN, self).init()
Hi, I am referring to your code and have some questions regarding forward function.
I am trying to apply VGG11 on CIFAR-10 dataset.
I applied the Average pooling layer after 2 Convolutional layers.
But in forward function, I am giving the spike of the first conv layer to the second conv layer and then applying avg pool layer to the second spike of conv layer. In this case, my code is not giving any output and error.
I don't know, what's wrong with this forward function and couldn't fix it.
Can you please tell me what should I do?
Your help will be appreciated.
Below is my code:
class SCNN(nn.Module): def init(self): super(SCNN, self).init()
in_planes, out_planes, stride, padding, kernel_size = cfg_cnn[0]