Ooooooooh you’re right I missed that, thanks a lot.
I fixed that and am getting some more mistakes(it truly seems like they never end).
ValueError: Target size (torch.Size([10])) must be the same as input size (torch.Size([10, 2]))

With a batch_size of 10, my output looks like this:
tensor([[0.0476, 0.0160],
        [0.0501, 0.0116],
        [0.0468, 0.0165],
        [0.0522, 0.0088],
        [0.0579, 0.0139],
        [0.0544, 0.0130],
        [0.0488, 0.0174],
        [0.0554, 0.0142],
        [0.0480, 0.0270],
        [0.0497, 0.0193]], grad_fn=<AddmmBackward>)

while my b_y looks like this:
tensor([0, 1, 0, 1, 1, 1, 1, 1, 0, 0])

I’m guessing that my output is the issue, but I’m not sure why it’s outputting two values per image or how to turn that into one value per image. Is there a function I’m missing? Or am I just not using the right loss function?