• 文档 >
  • torch.optim >
  • torch.optim.Optimizer.add_param_group
快捷键

torch.optim.Optimizer.add_param_group

Optimizer.add_param_group(param_group)[source][source]

Optimizer s 的 param_groups 中添加一个参数组。

这在微调预训练网络时可能很有用,因为冻结的层可以在训练过程中变为可训练并添加到 Optimizer 中。

参数:

param_group (dict) - 指定哪些张量应与特定优化选项一起优化。


© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,并使用 Read the Docs 提供的主题。

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源