torch.optim.Optimizer.add_param_group

Optimizer.add_param_group(param_group)

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

这对于微调预训练网络非常有用，因为冻结的层可以被设置为可训练，并在训练过程中添加到 Optimizer 中。

参数

param_group (dict) – 指定哪些张量应该被优化，以及组特定的优化选项。