bubbliiiing/hrnet-pytorch

关于ignore_label

littlezbq opened this issue · 1 comments

littlezbq commented

大佬好，我的数据集里面像素值为0的地方需要ignore掉，不参与loss的计算。我是不是在loss的相关函数里把ignore_index设置为0就可以了呢，我看您的代码那里默认的是num_class。

bubbliiiing commented

额0 0 也行……