zhoubolei/CAM

How to train CAM?

Opened this issue · 1 comments

if i want to train a cam which is used medical image(3 class),which loss is my need?

loss = torch.nn.MultiLabelSoftMarginLoss()?