yeah u were right it is supposed to be labels, labels is tensor, preds was already tensor.
But i have a doubt i rougly ran it for 1 epoch to see if the code for confusion matrix works…I had defined it in evaluation function…like on loading the model and then running the evaluation function on
train and test loader it should give some non- zero values but I’m getting all zeros only. What could be the error ?
Screenshot (1324)1443×497 9.82 KB