Hi, I’m currently using CrossEntropyLoss for a model with a single neuron as ouptut.
My predictions have the following pattern:
        [ 0.0445],
        [-0.1698],
        [-0.1343],
        [-0.5633],
        [ 0.2206],

To get my classes, do I have to do:
      pred_batch = torch.sigmoid(pred_batch)
      pred_batch[pred_batch >= 0.5] = 1
      pred_batch[pred_batch < 0.5] = 0

or
      pred_batch[pred_batch >= 0] = 1
      pred_batch[pred_batch < 0] = 0

Also, I’m getting some negative loss for my training set, is that normal?