When training three channel labels, an error occurred during running the validation set

[wangqingl]

How should I modify the code to resolve this error？