Shortcuts

torch.optim.Optimizer.add_param_group

Optimizer.add_param_group(param_group)[源代码]

Optimizerparam_groups 添加一个参数组。

这在微调预训练网络时非常有用,因为冻结的层可以变为可训练的,并随着训练的进行添加到优化器中。

Parameters

param_group (字典) – 指定应与组特定优化选项一起优化的张量。