叶妍青
叶妍青
My version: CUDA=9.0 Pytorch = 1.0.1 Apex installation appears to have been unsuccessful, the following error always occurs,I don't know what went wrong, please help me: /home/jingchengwang/anaconda3/lib/python3.5/site-packages/pip/_internal/commands/install.py:283: UserWarning: Disabling all...
您好,请问在训练的时候背景的Loss是不参与计算吗?因为我看您EL` Dice是从类别1开始计算的,如下: class EL_DiceLoss(nn.Module): def __init__(self, class_num=4,smooth=1,gamma=0.5): super(EL_DiceLoss, self).__init__() self.smooth = smooth self.class_num = class_num self.gamma = gamma def forward(self,input, target): input = torch.exp(input) self.smooth = 0. Dice = Variable(torch.Tensor([0]).float()).cuda()...