- 博客(16)
- 收藏
- 关注
原创 神经网络--线性层及其他层
跟着土堆一起学pytorch!正则化层不常用有的可以把数据变换为均值为 0、方差为 1 的分布这个用得比较多这些一般用于文字识别(建议看官方文档去学习)常用的为一下这些,橙色框里的为按需要学习讲解线性网络图解如下。
2025-03-13 15:35:20
513
原创 神经网络的基本骨架--nn.Module的使用
1.0就是一个0维度的张量,其实就是数字1,张量是pytorch中的各种维度数据形式的统称的,0维就是一个数,1维就是数组,2维是矩阵,3维是空间矩阵,...以此类推。先打开pytorch官网,点torch.nn就可以看到很多模型,其中containers是容器,里面的module是神经网络的最基本骨架。在写下面的def时,可以点上面的code,选override Methods,就可以自动弹出定义函数的格式。可以在第14行旁边点一下,出现红点点为断点,再点击右边的绿色小甲虫,可以实现逐步运行。
2025-03-06 07:52:29
528
原创 DataLoader的使用
改变时,如果tensorboard中最后一个step不足要求的一份batch_size=4的四张图片时,剩下的会被省去不显示。为什么我的tensorboard自从用来使用CIFAR10数据集时就用不了了,是因为我的电脑不是GPU吗?dataaset数据就像一张张扑克牌,而dataloard就像把牌抽出来,设置取多少,怎么取。可以先到pytoch官网中搜索dataloader的官方使用介绍,打开方法可以看我上一个文章。drop_last=False改为drop_last=Ture。运行后会出现图片的信息。
2025-03-04 13:12:18
385
原创 torchvision中的数据集使用(问题求助)
与transform结合,用compose整合函数后将PIL图片转为tensor类型,在transform=dataset_transform设置数据,选中文本后这里按Ctrl+/可全部注释,可以先运行看看数据类型有没有改变。先打开pytorch的官网,(最新版的在Search Docs左上角选择版本为0.9.0就一样了)点docs-pytorch,左边的pythonAPI减号去掉,点击library加号-torchvision便可以看到数据集。运行后会自动跳转展示img图片,显示是cat。
2025-03-02 14:37:17
747
原创 Pytorch Tutorial(一)
实际开始训练1.把模型调到训练模式(训练和测试模式做的事情会不一样)2.把dataset里的资料去处理3.把资料放到模型里面,给模型产生结果,再拿结果与真正和预期的结果去算loss,loss后会出现梯度,再用优化演算法去调整模型的参数,再历遍所有资料测试部分要把模型调到测试模式,再计算模型结果和loss的时候,把梯度计算关掉(跑得更快,如果代码写得不好会被测试资料来修正模型参数)保存和加载训练好的神经网络模型的方法。
2025-02-22 18:48:46
1069
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅