Error training data custom multitask_classification Tao-Toolkit

[_blackShark]

• Hardware: RTX Titan
• Network Type: Classification

I meet an mistake

Format data train:

[Morganh]

Please check the num of classes at train dataset and validation dataset. If they don’t match, there will be assertion error as above.

[_blackShark]

Thanks, I resolved it.