评价此页

torch.optim.Optimizer.add_param_group#

Optimizer.add_param_group(param_group)[来源]#

Optimizerparam_groups 添加一个参数组。

这在微调预训练网络时可能很有用,因为随着训练的进行,可以使冻结的层变得可训练并添加到 Optimizer 中。

参数

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