How to write custom Dataset to load hierarchical classes for feeding to a branched CNN in pytorch?

Question

I am trying to make an image classifier to classify the images into hierarchical classes (coarse_label1, coarse_label2, fine_label). The data is stored in a csv file with co