2 classification task at the same time
iumyx2612 opened this issue · comments
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?