ptrblck:

targets, class_counts = get_labels_and_class_counts(
                self.dataset.targets)



I have also fixed that issue, however problem is occuring when I inject that piece of code in my code. Moreover, could you please guide where (line number) have you made the above change.