Hi All,
I am doing a binary image classification using pytorch and I am using CNN architecture to do that.
In the final layer of architecture - do I need to give number of outputs parameter as “2” or “1”?  I placed as 1 and My final target columns have values [0,1,0,0,0,…]
Here is my code for the final layer:
        nn.Linear(128, 64),
        nn.ReLU(),
        nn.Linear(64, 1),
        nn.Sigmoid()
    )

Please clarify if it is correct. I am having some follow up questions on this.
–Shravan