yueyunqingwu

> 请问你训练时mask是什么样的呢？读取labels中的图片后经过ToTensor()处理的数值很小，然后转换为长整型（.long()），mask变为全0了，导致出现除数为0的情况。

我在使用pro模型推理时遇到该问题，定位在GLEE_Model类forward方法的返回值没有根据is_train这个布尔值进行区分，按理说is_train=False时只用return推理结果即可，稍微改一下代码即可，位置在projects/GLEE/glee/models/glee_model.py的359-370行。