Relax the train/val/test mask check for multi-task learning

Relax the train/val/test mask check for multi-task learning #5055