In retrospect, I don’t know if I can do this for training as the only way I could do it with validation was by splitting my validation set up into groups and running them through the network group by group… but the train loader randomly selects the data and so isolating the groups within a batch and then re-putting them through the network and cataloguing the accuracy seems very tricky