If I wanted to observe which data-points are being classified correctly at every epoch, is there a simple way to do this? I’m thinking I would get torch.max(net(inputs,1) but how would I obtain the indices of these data-points. Given that in:
max_vals, max_indices = torch.max(net(inputs,1)
max_indices actually returns the index of the correctly classified label and not the the index of the datapoint, is there a better way to do this?