Solved by justusschock in post #2 
              
              
                The following is working for me: 
class Discrim(nn.Module):
    channels, maxpool_mask = [13, 64, 192, 384, 256, 256], [1, 1, 0, 0, 1]
    ker_size, strd, pad = [2, 5, 3, 3, 3], [2, 1, 1, 1, 1], [0, 2, 1, 1, 1]


    def __init__(self, classes=13, conv_layers=5):
        super(Discrim, self).__init_…