pytorch
Dalier.
这个作者很懒,什么都没留下…
展开
-
pytorch
3.优化器:optimizer = torch.optim.SGD( 需要优化的参数 eg: model.paramenters(), lr = learning_rate)1.在定义模型时,def __init__(): 写一些你用到了了那些层。2.前向传播: 在上面写了这么多层,这些层怎么用的,怎么搭配。调用的时候:optimizer.step()5.创建 module 来更简化代码。4.把模型放到GPU上。原创 2024-03-16 17:33:41 · 362 阅读 · 0 评论 -
Dataset和DataLoader
是对数据的抽象,可以通过 Dataset 知道data 是一个多大的数据集,也可以通过他知道数据中某一行数据是什么。会在dataset之上会做进一步处理,将数据重新乱序,从中间拿出一个Batch出来。但是内存有限不能把数据一下子全给 module,所以就要分批次的输入数据。:这一Batch 数据,输入模型训练一遍就叫 Iteration。:将所有的 Iteration 都完成后,就叫一次Epoch。原创 2024-03-06 11:24:58 · 314 阅读 · 0 评论 -
Tensorboard的使用
Tensorboard原本是Google TensorFlow的可视化工具,可以用于记录训练数据、评估数据、网络结构、图像等,并且可以在web上展示,对于观察神经网络的过程非常有帮助。PyTorch也推出了自己的可视化工具,一个是tensorboardX包,一个是torch.utils.tensorboard,二者的使用相差不大,这里介绍后者。原创 2023-09-26 16:03:26 · 1069 阅读 · 1 评论 -
pytorch—torch.utils.data.Dataset 介绍与实战
【PyTorch】torch.utils.data.Dataset 介绍与实战_想变厉害的大白菜的博客-CSDN博客Pytorch学习—OS_Dalier.的博客-CSDN博客原创 2023-09-25 23:16:15 · 29 阅读 · 1 评论 -
Pytorch学习—OS
os.path模块主要用于文件的属性获取,exists是“存在”的意思,所以顾名思义,os.path.exists()就是判断括号里的文件是否存在的意思,括号内的可以是文件路径。os.listdir()函数:用于返回指定的文件夹包含的文件或文件夹的名字的列。os.path.join()用于路径拼接文件的路径。返回值:返回指定路径下的文件和文件夹列表。语法:os.listdir(path)参数:path – 需要列出的目录路径。原创 2023-09-25 22:16:18 · 46 阅读 · 1 评论