Like this (using PyTorch)? summed = 900 + 15000 + 800 weight = torch.tensor([900, 15000, 800]) / summed crit = nn.CrossEntropyLoss(weight ...
確定! 回上一頁