.modules:返回网络中所有网络信息,如卷积、线性等,还有索引。
代码示例:
import torch
import torch.nn as nn
l=nn.Sequential(nn.Conv2d(16,16,kernel_size=1),
nn.Conv2d(8,8,kernel_size=1))
print(list(l.modules()
输出:
[Sequential(
(0): Conv2d(16, 16, kernel_size=(1, 1), stride=(1, 1))
(1): Conv2d(8, 8, kernel_size=(1, 1), stride=(1, 1))
), Conv2d(16, 16, kernel_size=(1, 1), stride=(1, 1)), Conv2d(8, 8, kernel_size=(1, 1), stride=(1, 1))]