The output labels are 0, 1.  In the above code, the model output is
tensor([[-0.0190, -0.0389]], grad_fn=<AddmmBackward>)

and the target is
tensor([[0]])

The target data in data loader is in form
tensor([0., 0., 0.,  ..., 1., 1., 1.])