model.load_state_dict(state_dict, strict=False)

news/2024/7/8 6:32:57

我们在构造好了一个模型后,可能要加载一些训练好的模型参数。举例子如下:

假设  trained.pth 是一个训练好的网络的模型参数存储

model = Net()是我们刚刚生成的一个新模型,我们希望model将trained.pth中的参数加载加载进来,但是model中多了一些trained.pth中不存在的参数,如果使用下面的命令:

state_dict = torch.load('trained.pth')
model.load_state_dict(state_dict)

会报错,说key对应不上,因为model你强人所难,我堂堂trained.pth没有你的那些个零碎玩意,你非要向我索取,我上哪给你弄去。

但是model不干,说既然你不能完全满足我的需要,那么你有什么我就拿什么吧,怎么办呢?下面的指令代码就行了。

model.load_state_dict(state_dict, strict=False)

http://www.niftyadmin.cn/n/3657579.html

相关文章

新VB9出来了

原文:New VB9 available Amanda & Co 有新版编译器可供下载。它在DLINQ/选项中嵌入了XML,动态类型支持和对其功能。阅读相关信息,点击 这里。

VB9和Atom

原文: VB9 and Atom VB9和Atom我最近在试图说明C#所具有的高素养语言特点时,我的一大帮朋友都给我发来悲哀的信息。为了吸引我们中所有的软件蓝领,我想我应该用一小段VB9程序来清洗调色板,演示Erik & Co. 放在语言中对Xml的支…

conda: command not found

在终端输入conda info --envs检验anaconda是否安装成功,发现报错:conda: command not found 原因是因为~/.bashrc文件没有配置好 gedit ~/.bashrc 在最后一行加上 export PATH$PATH:/home/ansheng/anaconda3/bin ##写自己的对应路径 然后保存更改…

C#3.0/LINQ的新下载

原文: New drop of C# 3.0/LINQ 下载地址:http://www.microsoft.com/downloads/details.aspx?familyid1e902c21-340c-4d13-9f04-70eb5e3dceea&displaylangen我喜欢它的这一特色:LambdaExpression.CompileExpression> expr a > a 3;Console.…

C#中的代码和数据

原文:Code and Data in C# 我发现尽管我进了最大的努力来解释(中文blog链节),还是有不少家伙还是不知道LINQ是如何工作的。那么这回让我再试一次,速度稍慢一点,解剖一个简单的程序,来详细的讲解在Scheme 和C#3.0中的 c…

生成词云图片

import numpy as np from PIL import Image from wordcloud import wordcloud import jiebadef Product_WordClouds(text_file_path, background_picture, target_picture_path):"""函数说明:在调用这个函数的时候,传递三个参数&#xff…

Ruby,Block,CPS 和 call/cc

原文:Ruby, Blocks, CPS and call/cc这是一个在Chez Box有关Ruby的周末。这个夜晚我一直搞不懂,各种各样的原理是如何被应用得那么广泛,Ruby运行库又如何处理它们?下面,我希望了解: 传递/调用一个块与在当前…

Google搜索API?

原文: Google Search API? 有几个人给我讲了关于Google从基于SOAP访问机制的搜索转向基于AJAX的搜索。OReilly的文章开始对此进行了详细讲解,但是我认为这里还有更多的内容。我专门花了几分钟时间,在Google的站点上,将大家所指的…