欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

torch.nn modules

程序员文章站 2022-06-12 17:07:48
...

torch.nn

from torchvision.models import resnet18
net = resnet18()
for i, m in enumerate(net.modules()):
	print(i, m)

self.modules()是递归的

相关标签: Pytorch