yes, this method loads all the data into memory. If the data is large, I guess you can do this way. (I don’t try this)
def __getitem__(self, index):
    if self.train:
    shape_12v, label = self.modelnet40_data['train']['data'][index], self.modelnet40_data['train']['label'][index]

I don’t know if it works, you can tell me if you try.