2 classification task at the same time

[iumyx2612]

Hello I'm building a multi task classification, my model contains 1 feature extractor and 2 classification heads, so when implementing __getitem()__ it needs to return 1 image with 2 labels Is this any way to implement auto augment that way?