0
点赞
收藏
分享

微信扫一扫

关于我走过的弯弯绕绕

言诗把酒 2022-03-11 阅读 87

球球triplet成功吧

01测试时RuntimeError: CUDA out of memory

解决办法https://zhuanlan.zhihu.com/p/364332471

02预训练模型加载到更改后的网络中

load_state_dict方法有一个输入:strict,如果该参数为True,就表示网络结构的层信息要和预训练模型的层信息严格相等,反之亦然,该参数默认是True。那么什么时候会用到False呢?就是当你只想用预训练网络初始化你的网络的部分层参数或者说你的预训练网络的层信息和你要被初始化的网络的层信息不完全一致,那样就只会初始化层信息相同的层。

举报

相关推荐

0 条评论