在使用NLLLoss的时候,权重是根据target求的,但是有时候会遇到权重为nan,而当权重为nan的时候,NLLLoss的输出也就为nan,这导致后面loss全为nan了,而torch中的处理是权重为nan,loss输出则为0,然后不会影响后面 Environment / 环境信息 (Mandatory / 必填) Hardware Environment(Ascend/GPU/CPU) / 硬件环境: GPU Ple...
torch.nn.CrossEntropyLoss 最常用的多分类损失函数,内部原理如下 该损失函数的输入是yhat和y 对于一个样本来说 NN输出类别C个的logits, 记为向量y^ 样本的真实标签为y,标量,表示该样本属于第几个类别 计算logsoftmax(y^)值,是一个长度为C的向量,从中选取第y个元素,作为该样本的损失 ...
[MPS] Allow nan mean reduction in nll_loss #73054 Sign in to view logs Summary Jobs assign Run details Usage Workflow file Triggered via issue September 8, 2024 07:34 pytorch-bot[bot] commented on #135434 042f2f7 Status Success Total duration 12s Artifacts – assigntome-docathon.yml o...