because it’s a multilabel classification, you should use BCEWithLogitsLoss. The output of your network should come from a Linear layer and it should go into this loss.
http://pytorch.org/docs/0.3.0/nn.html#bcewithlogitsloss 1.3k