pytorch Module模塊學習

1、named_modules的使用

意思是 返回網路中所有模塊的迭代器，同時產生模塊名和模塊本身。

Note中的意思是重複模塊之返回一次。

看了example還是沒懂這個意思是什麼，所以做了一下實驗。

import torchimport torch.nn as nnl = nn.Linear(2, 2)k = nn.Linear(3, 3)net = nn.Sequential(l, k)for idx, m in enumerate(net.named_modules()): print(idx, ->, m)

可以看到網路由l，k組成構成Sequential，但是返回結果還包括了序列中的每部分。所以就可以根據id去調用每個子模塊。

推薦閱讀：