2019年7月24日
怎么仅加载一部分内容的预训练模型参数
摘要： 在pytorch中提供了很多预训练好的模型，以分类为例，基本上都是用ImageNet数据集来训练的，分为1000类。 但是很多时候我们要实现的分类项目可能并没有这么简单，比如我们可能并不仅仅只是实现单分类，可能想实现双分类或者是多分类，这个时候就需要对模型进行一定的修改 修改的同时还希望该修改后的模 阅读全文
posted @ 2019-07-24 10:24 慢行厚积 阅读(6880) 评论(0) 推荐(1) 编辑