2021年3月31日
pytorch nll_loss与CrossEntropyLoss示例
摘要： nll_loss 输入的是求过log_softmax之后的值，默认reduction='mean'，计算的是平均loss，即将targets对应的log_prob相加再求均值： >>> import torch >>> logits=torch.randn(2,3) >>> logits tenso 阅读全文
posted @ 2021-03-31 19:26 AliceYing 阅读(408) 评论(0) 推荐(0)