Greedy Hash如何在多标签数据NUS-WIDE, COCO上训练

[ssqiao]

trafficstars

非常感谢你的project，实在太棒了。有个问题请教，Greedy Hash是基于交叉熵目标分类损失，但是NUS和COCO数据集是多标签图像，如何训练Greedy Hash，看到你给了结果，但不知道怎么训练？

[swuxyj]

也是用的torch.nn.CrossEntropyLoss()，这两个数据集用相同的流程也能跑出一个结果出来

[ssqiao]

应该是BCELoss吧，普通的CrossEntropy似乎不能处理多标签分类问题，只能处理单标签分类