Hello,
when I’m trying to return predicted values with following code:
preds = y_val.to('cpu').data
prediction = torch.max(preds, 1)[1].numpy()

this only returns 7 predicted labels, i don’t get why exactly 7, when there should be almost 6000 labels. How to return the full amount of predicted labels?