- 博客(6)
- 收藏
- 关注
RSS订阅原创 完整的模型训练套路////(b站小土堆视频学习笔记)
网络模型可以单独放在一个Python文件中，只需在原来的文件中import一下，例如将网络模型的Python文件名命名为model，在原来的Python文件中 from model import *二、利用DataLoader,来加载数据集。可清晰地看到训练跟测试的损失都在 减小。返回64行数据， 每行数据有10个。注意这两个文件需要在同一个路径下。网络模型下面可以测试网络的正确性。用tensorboard可视化。准备了数据集并加载了长度。七、设置训练网络的一些参数。
2024-08-07 20:46:45
194
原创 网络模型的保存与加载///(b站小土堆学习笔记）
如果要恢复网络结构，新建一个网络结构vgg16，vgg16加载其中的参数，通过字典的形式，方式1 保存了网络模型的结构，也保存了网络模型的参数。但是当加载时， 报错找不到Tudui这个类。加载模型后打印出的是字典形式。运行成功后，会出现如下文件。此时用 方式1有一个陷阱。需要将定义的类粘贴在上面。或者前面import一下。方式2，保存了模型参数。这是保存后，没有问题。
2024-08-06 14:59:46
123
原创 对现有的网络模型的添加与修改|||（b站小土堆学习笔记）
最后 一个线性层输出1000，我们用CIFAR10最后是把数据分为10类，有两种解决办法，第一种将最后的线性层的输出out_features=10，第二种在最后一个线性层后面加一个输入为1000，输出为10。新版本中这个参数被取代为weights，如果不进行预训练，在weights = none；pretrained如果为True的话，相当于下载的网络模型的一些参数已经在ImageNet这个数据集当中训练好了，在这个数据集当中能取到一个不错的效果。报错此数据集不能公开访问，需下载。
2024-08-02 11:10:59
351
原创 Transforms的使用/// 小土堆学习笔记
从transforms中选择一个class创建，根据创建的工具看需要什么东西，然后返回一个结果，可以看出图片类型由PIL转换成tensorCtrl+p 查看需要什么参数，这里显示pic，即需要一个图像参数Ctrl+p 查看需要什么参数，这里显示pic，即需要一个图像参数。tensor数据类型可以理解为，它包装了神经网络所需要的理论基础的一些参数，几乎所有的都需要转换成tensor ①确保数据格式的一致性，便于模型处理②变为浮点数③归一化，像素值通常在0到255之间，通常归一化到[0,1]或[
2024-07-22 17:16:39
1373
1
原创 TensorBoard的使用 //（小土堆学习笔记）
导入（从torch.常用的工具箱中导入tensorboard,导入SummaryWriter类）pycharm中按住Ctrl可查看属性等。
2024-07-20 17:19:18
667
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人