Solved by ptrblck in post #28 
              
              
                That’s good to hear! 
Is the initialization working now? 
This code runs perfectly on my machine: 
import torch
import torch.nn as nn

class CNN(nn.Module):
    
    def __init__(self, out_1=13, out_2=32):
        super(CNN, self).__init__()
        self.cnn1 = nn.Conv2d(in_channels=3, out_channels=…