摘要: #1. torch.nn.CrossEntropyLoss() 用于多分类,本身含有softmax #2. torch.BCELoss() 用于二分类,本身不含有sigmoid,需要在网络里面自己增加 #3. torch.BCEWithLogitsLoss() 用于二分类,相当于 BCELoss+s 阅读全文
posted @ 2021-12-19 21:05 骑猪猪看日出哦 阅读(120) 评论(0) 推荐(0)