If I use nn.BCEWithLogitsLoss() instead, what is the proper input of the nn.BCEWithLogitsLoss() ? Do I have to scale the output value of the NN to be 0 to 1 (i.e., Softmax function)? Or do I need other activation functions… ?
Also, what about nn.CrossEntropyLoss()  ? What is the proper input for nn.CrossEntropyLoss(). ?