深度学习
文章平均质量分 58
哒哒哒的龟
这个作者很懒,什么都没留下…
展开
-
pytorch 加载 state_dict
在Pytorch中,一个模型的可学习参数(learnable parameters),就是权重以及偏置可以通过以下代码获取state_dict就是一个python字典对象,该对象里面储存的是每一层所对应的可学习参数(learnable parameters) ...原创 2022-08-31 14:22:35 · 1131 阅读 · 0 评论 -
论文阅读:CNN: Single-label to Multi-label(未完待续)
@[TOC](论文阅读:CNN: Single-label to Multi-label (未完待续))论文地址: https://arxiv.org/pdf/1406.5726.pdfAbstract众所周知CNN能在单标签分类任务上表现的很好,但是,CNN如何识别多标签图像,对多标签图像进行分类仍然是一个问题,主要是由于物体的布局还有多标签训练图像不足。在这篇论文作者提出了一种新的结构,Hypotheses-CNN-Pooling (HCP)。任意数量的假设目标存在的部分会作为HCP的输入,所有的原创 2021-02-04 12:07:59 · 237 阅读 · 2 评论 -
Pytorch加载训练好的模型并预测
Pytorch加载训练好的模型并预测Pytorch模型的保存模型的加载利用加载好的模型预测(照片)Pytorch模型的保存根据官方介绍,pytorch模型的保存有不同的方法,这里使用的是官方推荐的只保存模型的state_dict(weight and bias).# model: 模型名称 # save_path: 模型保存的路径torch.save(model.state_dict(), save_path)模型的加载官方给的代码如下:# the model class: 你所定义的模型原创 2021-02-03 18:25:51 · 25347 阅读 · 15 评论