问题:图片大小为1024x1024,第一个batch跑完的时候很正常,到了第二个batch以及之后的都是nan 可能的原因:
暂时想到这些,会来补充并填坑!
2020年9月5更新: 我最终loss为nan的原因是聚类那边写错代码了,我修改之后效果很好!!