ENVI5.6深度学习模块环境测试出错，应该如何解决呢？

在最下面测试Training a Tensorflow Model出错，显卡部分通过

看了下报错信息，是
ERROR:Losscalculationfailed
错误：损失计算失败
ERROR:Modelcould not betrained
错误：无法训练模型

建议看下这个思路：
loss的计算中不能出现自乘操作
pred_mask = pred * mask （对）
pred *=mask (错)