PyTorch学习
文章平均质量分 71
记录本人学习pytorch过程,共勉!
一步一个脚印ッ
从事联邦学习相关研究
展开
-
PyTorch实战GAN生成Mnist数据集
使用GAN生成Mnist数据集,数据不需要你提前准备,运行gan.py自动下载数据。进行训练,打印训练过程,保存生成的Mnist图片到images文件夹下。经过100个周期后生成的图片,已经可以看出来7,9了。图像形状,输入通道1(黑白),28*28(长宽)查看结果,最开始生成的图片,几乎无法辨别。即下面的BCEloss.py。设定Mnist图片生成路径。实例化生成器与判别器。原创 2022-11-14 20:12:04 · 1493 阅读 · 0 评论 -
PyTorch实战CNN文本分类
CNN原本用于图像任务,如何利用CNN进行文本分类任务?关键在于特征提取方式,进过上一节,我们知道LSTM利用了词嵌入模型,将词映射为一个300维的向量。这里我们也可以利用同样的方式,举例:先将一个字映射为5维向量,设卷积核大小为4。5(绿色部分)或者2*5(黄色部分),颜色深浅为两个卷积核即两个初始化权重参数不同的卷积核。最大池化max-pooling提取特征点,进行拼接,再进行全连接进行分类。全部代码在上一节LSTM中,代码实现了LSTM模块和CNN模块,注意需要把model默认值改为TextCNN。原创 2022-11-08 20:04:00 · 1205 阅读 · 0 评论 -
PyTorch实战LSTM新闻分类
PyTorch实现LSTM进行新闻分类任务,数据集代码公开获取。原创 2022-11-08 16:27:45 · 1571 阅读 · 0 评论 -
PyTorch实战使用Resnet迁移学习
项目数据集:102种花的图片。项目算法:使用迁移学习Resnet152,冻结所有卷积层,更改全连接层并进行训练。原创 2022-10-11 22:13:25 · 2014 阅读 · 1 评论 -
PyTorch实战mnist图像分类
使用pytorch进行mnist图像分类原创 2022-10-09 20:08:57 · 648 阅读 · 0 评论 -
PyTorch实战气温预测
本次任务是进行气温预测temps.csv,数据集文章中自取,作为全面学习PyTorch实战的第一章,我们会使用比较原始的方法写整个训练过程,除了反向传播由PyTorch代码调用自行计算。原创 2022-10-08 09:47:46 · 2688 阅读 · 4 评论