classification-pytorch
classification-pytorch copied to clipboard
你好,这个任务修改之后可以完成多标签多类别 分类的任务吗
如题
额0 0你的意思是同时输出多个标签?
sigmoid?
额0 0你的意思是同时输出多个标签?
比如同时输出一张猫图像为“ 猫 ”和 “动物”,是不是可以采用类似下面的方法 `out_1 = self.fc_1(x)
out_2 = self.fc_2(x)
return out_1, out_2
在模型训练时,计算多个分类损失相加
loss = criterion(pred1, y1) + criterion(pred2,y2)`
额0 0你的意思是同时输出多个标签?
比如同时输出一张猫图像为“ 猫 ”和 “动物”,是不是可以采用类似下面的方法 `out_1 = self.fc_1(x)
out_2 = self.fc_2(x)
return out_1, out_2
在模型训练时,计算多个分类损失相加
loss = criterion(pred1, y1) + criterion(pred2,y2)`
我现在确实是这样弄的