model.add_module
Pytorch使用這個方法,就可以在已有模型的基礎(chǔ)上重新加一些結(jié)構(gòu),比如已經(jīng)加載了某個預(yù)訓(xùn)練模型,但仍需要增加一些模塊,此時已經(jīng)不能在一開始構(gòu)建模型的時候init進(jìn)去模塊,于是可以在load完模型后,用model.add_module,把所需要的模塊增添進(jìn)去,使用方法是model.add_module("module_name", nn.Sequential(...))
但需要注意的一點是,如果load進(jìn)來的模型在GPU上了,就要先model.cpu(),然后再add_module,之后再放GPU上。
人生苦短,何不用python

浙公網(wǎng)安備 33010602011771號