model.apply(fn)或net.apply(fn)
首先,我们知道pytorch的任何网络net,都是torch.nn.Module的子类,都算是module,也就是模块。pytorch中的model.apply(fn)会递归地将函数fn应用到父模块的每个子模块submodule,也包括model这个父模块自身。比如下面的网络例子中。net这个模块有两个子模块,分别为Linear(2,4)和Linear(4,8)。函数首先对Linear(2,4)和Linear(4,8)两个子模块调用init_weights函数,即print(m)打印Linear(2,4
原创
2020-06-13 23:18:57 ·
19122 阅读 ·
14 评论