search

kitsch231 / pytorch_fake_news_Classification_mml

使用pytorch完成的一个多模态分类任务，文本和图像部分分别使用了bert和resnet提取特征（在config里可以组合多种模型）,在我的小规模数据集上取得了良好的性能（验证集acc96%）
61 stars 8 forks source link

models里面没必要做logits=self.softmax(logits) #7

Open leixingzhi7 opened 1 month ago

leixingzhi7 commented 1 month ago

models里面没必要做logits=self.softmax(logits)，本身输出就是logits，后面用的是CrossEntropyLoss，里面会做softmax，虽然结果不影响，但是这步骤确实是多余了

kitsch231 commented 4 days ago

具体原因忘记了，可能是我当时尝试了很多的loss函数和模型结构来改进精度，不过目前没有其他影响。