Sort of worked but having issues with the target variable, I thought by setting the images on folder 1/2/3 respectively would’ve loaded the target correctly.
I followed some of the other comments you had for other people on how to unsqueeze the data and so on, But my target comes out of the loader as an int, not an an ‘I guess’ hot encode PyTorch tensor?
if I print right after input, i get a torch.Size([1, 256, 256]) and the output is just a integer.
I did some transformations, but i think i am doing the wrong thing because my train loss is fixed.
Why are the labels not loaded properly as a long encode when using image folder?
I did this:
    inputs = inputs.unsqueeze(0)
    labels = torch.tensor(labels).unsqueeze(0)

Ian
But i think the shape of the target is not right.