pytroch学习
文章平均质量分 71
菜菜子-
这个作者很懒,什么都没留下…
展开
-
遇到的pytorch不懂的地方
一些备忘录原创 2023-02-12 21:39:06 · 114 阅读 · 0 评论 -
pytorch学习记录四【cifar示例完整代码 / 利用GPU进行训练 / 测试】
import torchimport torchvisionfrom tensorboardX import SummaryWriterfrom torch import nnfrom torch.utils.data import DataLoaderfrom cifar_src.model import *# 准备数据集train_data = torchvision.datasets.CIFAR10("../cifar_data", train=True, transform=torc原创 2022-02-12 17:07:34 · 1160 阅读 · 0 评论 -
pytorch学习记录三 【Sequential / 损失函数 / 反向传播 / 优化器 / 模型修改及使用 / 模型保存与加载】
1.神经网络:搭建小实战和Sequential的使用以CIFAR10 model为例import torchimport torchvisionfrom tensorboardX import SummaryWriterfrom torch import nnfrom torch.nn import ReLU, Sigmoid, Linear, Conv2d, MaxPool2d, Flattenfrom torch.utils.data import DataLoaderclas原创 2022-02-11 20:22:22 · 1168 阅读 · 0 评论 -
pytorch学习记录二 【数据集使用 / DataLoader / nn.Module / nn.conv / conv2d / maxpool / ReLU / linear】
1.torchvision中的数据集使用▲pytorch官网官网链接:https://pytorch.org/进入torchvision在Package Reference中torchvision.datasets :torchvision中给的一些数据集,例如COCO等。继续往下滑▲torchvision.datasets以cifar-10为例import torchvisiontrain_set = torchvision.datasets.CIFAR10(root=".原创 2022-02-11 11:09:17 · 1207 阅读 · 0 评论 -
pytorch学习记录一【搭建环境/pycharm/jupyter/Dataset/TensorBoard/transforms】
▲conda命令: conda命令详解1.创建环境例子: #创建python3.6版本的环境test: conda create -n test python=3.6#激活该环境test:conda activate test#展示该环境的包:pip list2.确定cuda和gpu▲电脑查询GPU型号:任务管理器-性能or设备管理器-显示适配器▲电脑查询GPU型号对应的cuda:命令行输入:nvidia-smiorNVIDIA控制面板-系统信息-组件-3D设置-NV原创 2022-01-20 16:08:34 · 1400 阅读 · 0 评论