
PyTorch中torch.nn模块的损失函数
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文介绍了PyTorch框架中的torch.nn模块,重点讲解了其中的各种损失函数及其在深度学习模型训练中的应用。
前言
最近在使用Pytorch进行多标签分类任务时遇到了一些关于损失函数的问题。由于经常容易忘记这些细节(毕竟好记性不如烂笔头),我习惯于现学现用,因此自己编写了一些代码来探究这些问题,并在此记录下来以备将来参考。如果以后还遇到其他的损失函数,我会继续补充相关信息。
一、torch.nn.BCELoss(weight=None, size_average=True)
二、nn.BCEWithLogitsLoss(weight=None, size_average=True)
三、torch.nn.MultiLabelSoftMarginLoss(weight=None, size_average=True)
四、总结
全部评论 (0)
还没有任何评论哟~


