model.add_module


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

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


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM