raise RuntimeError(‘Error(s) in loading state_dict for {}:\n\t{}‘.format( RuntimeError: Error(s) in
一个pytorch使用过程中的错误加载预训练权重的过程中因为没使用VGG的其他的一些层的预训练权重,bias。产生了一个错误我们只需要load_state_dict(fsd,strict=False) 即可加一个strict=False这个参数属性strict;当strict=True,要求预训练练权重层数的键值与新构建的模型中的权重层数名称完全吻合;如果我们采用strict=False就能够完美的解决这个问题。也即,与训练权重中与新构建网络中匹配层的键值就进行使用,没有的就默认初始化。..
原创
2021-12-01 10:56:59 ·
4443 阅读 ·
1 评论