torch.optim.Optimizer.add_param_group

Optimizer.add_param_group(param_group)[源代码]

将一个参数组添加到Optimizerparam_groups中。

在微调预训练网络时,这非常有用,因为可以将冻结的层设为可训练状态,并随着训练进度将其添加到Optimizer中。

参数

param_group (dict) – 指定需要优化的张量及其特定的优化选项。