Solved by ayalaa2 in post #8 
              
              
                This is getting closer, but that conditional is still throwing me off. I’ll use the network described in your message. 
criterion = nn.BCELoss()
encoder_net = Encoder(input_size, hidden_size, output_size)
classifier_net = Classifier(2 * output_size, hidden_size)  # I'm allocating room for 2 tensors …