PyTorch快速入门
快速学习pytorch基本模块
千禧霓虹
这个作者很懒,什么都没留下…
展开
-
(十六)完整的模型训练套路
(9)数据可视化与模型保存。(6)设置训练网络参数。原创 2024-01-20 20:07:45 · 453 阅读 · 0 评论 -
(十五)网络模型的保存与加载
【代码】(十五)网络模型的保存与加载。原创 2024-01-19 20:53:47 · 366 阅读 · 1 评论 -
(十四)现有网络模型的修改
【代码】(十四)现有网络模型的修改。原创 2024-01-19 20:23:03 · 796 阅读 · 1 评论 -
(十三)优化器
Module):my_module = My_Module() loss = nn . CrossEntropyLoss() # 定义损失函数optimizer = optim . SGD(params = my_module . parameters() , lr = 0.01) # 定义优化器 , 采用随机梯度下降算法。原创 2024-01-19 19:34:09 · 367 阅读 · 1 评论 -
(十二)损失函数与反向传播
【代码】(十二)损失函数与反向传播。原创 2024-01-19 18:53:19 · 377 阅读 · 1 评论 -
(十一)神经网络- CIFAR-10的网络模型搭建及Sequential的使用
dilation默认值是1,代表不使用空洞卷积。原创 2024-01-19 11:22:47 · 429 阅读 · 1 评论 -
(十)神经网络-线性层
实现下图Fully Connected的类似部分(图中数据与代码无关)Module):Linear(in_features=196608, out_features=10) # 使torchSize([10])imgs_tensor = torch.flatten(imgs) # torch.flatten()作用是将张量展平,使torch。原创 2024-01-18 22:06:17 · 444 阅读 · 1 评论 -
(九)神经网络-非线性激活
【代码】(九)神经网络-非线性激活。原创 2024-01-18 21:19:43 · 380 阅读 · 1 评论 -
(八)神经网络-池化层
stride = 3。原创 2024-01-18 20:46:59 · 352 阅读 · 1 评论 -
(七)神经网络-卷积层
卷积演示动态图3.Convolution Layers3.1 torch.nn.Conv2d类介绍原创 2024-01-18 11:22:48 · 367 阅读 · 1 评论 -
(六)神经网络的基本骨架-nn.Module的使用
子类继承nn.Module需要重写forward()方法。原创 2024-01-18 10:07:24 · 412 阅读 · 1 评论 -
(五)DataLoader的使用
writer . add_images(tag = "test_loader" , img_tensor = imgs , global_step = step) # 因为batch_size = 64 , 所以使用add . images() 函数而不是add . image() step += 1 writer . close()ToTensor()])step = 0# 因为batch_size=64,所以使用add.images()函数而不是add.image()step += 1。原创 2024-01-17 22:59:33 · 519 阅读 · 0 评论 -
(四)torchvision中数据集的使用
1.CIFAR-10数据集介绍2.torchvision内含数据集的下载from torchvision import datasetsfrom torchvision import transformsfrom torch.utils.tensorboard import SummaryWritercompose = transforms.Compose([ transforms.ToTensor()])train_dataset = datasets.CIFAR10(root=原创 2024-01-17 22:12:24 · 382 阅读 · 0 评论 -
(三)Transforms的使用
Compose()中的参数需要是一个列表,Python中列表的表示形式为[数据1,数据2,数据3,…在Compose中,数据需要是transforms类型,所以得到Compose([transforms参数1,transforms参数2,…原创 2024-01-17 19:24:35 · 394 阅读 · 0 评论 -
(二)TensorBoard的使用
tensorboard --logdir=logs – port=6007,指定端口。tensorboard --logdir=logs,logs是事件文件所在的文件夹。默认浏览器要改成Google,safari不支持。原创 2024-01-17 17:36:45 · 417 阅读 · 0 评论 -
(一)DataSet类
【代码】(一)DataSet类。原创 2024-01-17 11:32:46 · 370 阅读 · 0 评论