获取分块网络层–使用nn.Module的children()方法实现
for k ,layer in enumerate(model.children()): #将model替换为你自己的模型名字
print('第%d层如下:'%(k+1))
print(layer)
获取分块网络层名称–使用nn.Module的named_children()方法实现
for k ,(name,layer) in enumerate(model.named_children()): #将model替换为你自己的模型名字
print('第%d层如下:'%(k+1,name))
print(layer)
获取所有网络层–使用modules()方法实现
for k ,layer_block in enumerate(model.modules()): #将model替换为你自己的模型名字
print('第%d层'%(k+1))
print(layer_block)