Trying to implement a recurrent Resnet of sorts…
Stuck on this error…
Please help this lost lamb
class Block(nn.Module):
    def __init__(self, channels, seq):
        super(Block, self).__init__()
        self.seq = seq
        self.channels = channels
        self.conv = nn.Conv2d(self.channels, self.channels, 3, padding = 1, stride = 1, groups = self.seq)
        self.bn = nn.BatchNorm2d(self.channels)
        self.relu = nn.ReLU()
        
    def forward(self, tensor):

        identity = tensor
        tensor = self.conv(tensor)
        tensor = self.bn(tensor)
        tensor = self.relu(tensor)
        tensor = self.conv(tensor)
        tensor = self.bn(tensor)

image1062×477 13.6 KB