import torch
conv = torch.nn.Conv2d(1,8,(2,3))
bn = torch.nn.BatchNorm2d(8)
l = [conv,bn]
for item in l:
print(item.__class__.__name__)
输出结果:
Conv2d
BatchNorm2d
import torch
conv = torch.nn.Conv2d(1,8,(2,3))
bn = torch.nn.BatchNorm2d(8)
l = [conv,bn]
for item in l:
print(item.__class__.__name__)
输出结果:
Conv2d
BatchNorm2d