随笔分类 -  Pytorch笔记

记录使用pytorch的笔记
pytorch加载已下载的FashionMNIST数据集
摘要：在获取数据集时，加载本地的，download为false mnist_train = torchvision.datasets.FashionMNIST(root=root, train=True, download=False, transform=transform) mnist_test = 阅读全文
posted @ 2022-03-01 16:14 骑猪猪看日出哦 阅读(511) 评论(0) 推荐(0)
Pytorch:三个交叉熵损失函数区别
摘要：#1. torch.nn.CrossEntropyLoss() 用于多分类，本身含有softmax #2. torch.BCELoss() 用于二分类，本身不含有sigmoid，需要在网络里面自己增加 #3. torch.BCEWithLogitsLoss() 用于二分类，相当于 BCELoss+s 阅读全文
posted @ 2021-12-19 21:05 骑猪猪看日出哦 阅读(111) 评论(0) 推荐(0)