会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
慢行厚积
博客园
首页
新随笔
联系
订阅
管理
2019年7月24日
怎么仅加载一部分内容的预训练模型参数
摘要: 在pytorch中提供了很多预训练好的模型,以分类为例,基本上都是用ImageNet数据集来训练的,分为1000类。 但是很多时候我们要实现的分类项目可能并没有这么简单,比如我们可能并不仅仅只是实现单分类,可能想实现双分类或者是多分类,这个时候就需要对模型进行一定的修改 修改的同时还希望该修改后的模
阅读全文
posted @ 2019-07-24 10:24 慢行厚积
阅读(6880)
评论(0)
推荐(1)
编辑
公告