model.add_module

Pytorch使用这个方法,就可以在已有模型的基础上重新加一些结构,比如已经加载了某个预训练模型,但仍需要增加一些模块,此时已经不能在一开始构建模型的时候init进去模块,于是可以在load完模型后,用model.add_module,把所需要的模块增添进去,使用方法是model.add_module("module_name", nn.Sequential(...))

但需要注意的一点是,如果load进来的模型在GPU上了,就要先model.cpu(),然后再add_module,之后再放GPU上。

posted @ 2021-03-12 09:41  嶙羽  阅读(1096)  评论(0编辑  收藏  举报