pytorch
LightningJie
这个作者很懒,什么都没留下…
展开
-
pytotch(八):非线性激活
inplace: 是否更改传入的数据。例如 input=-1 ReLU(input,inplace=True) 此时,input的值变为0, input=-1 output=ReLU(input,inplace=False) 此时,input的值依然为-1,output的值为0.原创 2024-07-18 21:03:57 · 191 阅读 · 0 评论 -
pytorch学习(七):池化层的使用
在池化层中,假设kernel_size=3,也就是3*3的大小,但是原始图片为5*5,这样在进行第二次池化的时候,窗口里只有6个元素了,不够3*3了,这样会根据ceil_mode是floor和Ceiling模式来进行保留或者舍去。解释一下卷积层和池化层的区别:卷积的作用是提取图像特征,池化的作用是保留边缘特征,减少参数,池化分为最大池化(保留边缘特征)、平均池化(保留背景特征)。stride:窗口的步长,默认值是kernel_size的值。dilation:如下图,控制窗口内内元素之间的距离。原创 2024-07-17 11:59:48 · 181 阅读 · 0 评论 -
pytorch学习(六):卷积层的使用
s=1。原创 2024-07-16 20:04:02 · 332 阅读 · 0 评论 -
pytorch学习(五):搭建神经网络模型
forward函数在任何子类中都需要重写!原创 2024-05-19 13:36:42 · 138 阅读 · 0 评论 -
pytorch学习(四):Dataloader使用
遇见问题:writer.add_image()报错。报错原因:应该都是write.add_images()dataloader:加载器,把数据加载到神经网络当中。从dataset中取数据。write.add_images() 是添加多张图片。dataset:读取数据。原创 2024-05-18 14:20:30 · 196 阅读 · 0 评论 -
pytorch学习(三):torchvision中的数据集的使用
【代码】pytorch学习(3):torchvision中的数据集的使用。原创 2024-05-16 11:19:55 · 128 阅读 · 0 评论 -
pytorch学习(二):transforms使用
作用:将特定的图片通过transforms工具处理,得到我们想要的结果。原因:在PyTorch的模块中,ToTensor是一个用于图像预处理的类,而不是一个函数。因此,你不能直接调用来转换图像,因为这不是这个类设计的使用方式。ToTensor类是用来创建一个预处理对象的,该对象可以被调用以将PIL图像或NumPy ndarray转换为PyTorch张量(tensor)。ToTensor遇见小错误:tensorboard无法显示图片,原因:忘记 writer.close()在Python的类中,原创 2024-05-15 13:55:51 · 397 阅读 · 0 评论 -
pytorch学习(一):tensorboard使用
注意问题:pillow的版本要求在10.0以下,否则会如下报错:module 'PIL.Image' has no attribute 'ANTIALIAS'简单画一个y=x图像模拟模型训练时候的损失函数的图像。在终端输入下方指令,点击连接即可查看图像。安装后,就可以使用tensorboard;原创 2024-05-12 19:18:25 · 207 阅读 · 0 评论