代码之家 › 专栏 › 技术社区 › mrgloom

参数和子参数有什么区别?

pytorch

mrgloom · 技术社区 · 6 年前

看起来像 parameters 和 children 显示相同的信息,那么它们之间的区别是什么?

import torch

print('torch.__version__', torch.__version__)

m = torch.load('imagenet_resnet18.pth')

print(m.parameters)
print(m.children)

1 回复 | 直到 6 年前

backtothemoon 6 年前

model.parameters() 是返回包含模型参数的张量的生成器。
model.children() 是一个返回模型层的生成器,从中可以使用 <layername>.weight 或 <layername>.bias

访问此 link 有关访问和冻结模型层的简单教程。

推荐文章

Saffy · 如何在IterableDataset上应用最小最大缩放?

5 月前

sanjeev mk · 通过索引从Pytorch或Numpy 2D数组中快速删除多行的方法

1 年前

Anonymous · 如何为零维火炬张量赋值?

1 年前

JohnnyWang97 · getattr引起的有趣错误

1 年前

Kamugg · 在PyTorch中使用不同分辨率图像训练DeepLabV3的最佳实践

1 年前

Stocavista · 无法在python中将float 64转换为float 32

1 年前

efwefwefwefwefw wefwefwefwef · 如何在PyTorch Conv1d层中仅在一侧应用填充?

1 年前

Okhr · 运行时错误:CUDA错误:在带有GTX 1660 Super的Debian 12虚拟机上不支持此操作

1 年前

Fatemeh · 如何从使用nn训练和保存的模型加载检查点。DataParallel到不使用nn的模型上。DataParallel?

1 年前

Twenkid · 将GPT2 h5型号转换为割炬,以转换为ggml形状不匹配

1 年前