result type Float can't be cast to the desired output type Long

[alian123456]

训练过程中显示类型转换错误，为torch版本问题，目前所用版本为torch1.7.0+cu110，请问这个问题有遇到过吗？