学习笔记
文章平均质量分 64
文山湖的猫
Long may the Sun shine!
展开
-
pytoch使用nn.DataParallel导致Missing key(s) in state_dict错误
1.nn.DataParallelnn.DataParallel是pytorch使用多gpu训练时所使用的方法,但是使用nn.DataParallel之后,模型的读取就会有所不同。最常见的情况就是使用的预训练模型并不是在多张gpu上训练得来的,没有使用nn.DataParallel包装,但现在想要使用这个预训练模型就不能直接用下面的代码model.load_state_dict(torch.load(save_path))否则就会报以下错误Missing key(s) in state_dict:原创 2021-03-05 21:09:26 · 1610 阅读 · 1 评论 -
vscode 出现“identifier “cout“ is undefined“或“未定义标识符cout”
vscode 出现"identifier “cout” is undefined"或“未定义标识符cout”最近想学C++,然后就从通读C++ Primer Plus开始,结果出师未捷身先死,第一个例程就报错,识别不了cout。中文网上搜索到的大部分结果都是说加上using namespace std;但我的代码里本身就有这一句。最后还是谷歌靠谱,应该是vscode的bug。File-》Preference-》Setting-》Extensions-》C/C++"C_Cpp.intelliS原创 2020-12-13 12:03:52 · 10480 阅读 · 21 评论