Pytorch使用這個方法,就可以在已有模型的基礎上重新加一些結構,比如已經加載了某個預訓練模型,但仍需要增加一些模塊,此時已經不能在一開始構建模型的時候init進去模塊,於是可以在load完模型后,用model.add_module,把所需要的模塊增添進去,使用方法是model.add_module("module_name", nn.Sequential(...))
但需要注意的一點是,如果load進來的模型在GPU上了,就要先model.cpu(),然后再add_module,之后再放GPU上。