2301_78018149的博客

原创 CLIP模型进行图像分类时，遇到了nan情况。

大家好，最近我在用CLIP模型去对图片进行分类并输出每个类大的概率值。但是，在训练时出现了每一个epoch的loss值都是nan的情况，也尝试着去降低学习率，对数据集进行统一等等方法去解决。但是目前还是每一个epoch都是nan，有大佬知道还有什么方法去解决嘛？代码我放在下面啦，数据集里面我对图像进行了简单的4分类，每一类里面只有10张图片。之前运行的时候是成功的，而且准确率也特别高，但是后来这个代码也没动，再次去运行的时候就会出现nan。真诚期待大家的回复！