在源代码的loss.py中有这么一句
indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0,gain[2]-1)))
直接训练的话汇报数据类型的错误,大概的意思是浮点数不能转换成长整型,因为indices是个只能存放整数的数组,具体的原因是在最新版的torch中clamp_方法改过了,现在要改成clamp,在加上long()确保是整型数据:
indices.append((b, a, gj.clamp(0, gain[3] - 1).long(), gi.clamp(0,gain[2]-1).long()))