megengine.optimizer.AdamW.add_param_group

AdamW.add_param_group(param_group)

向 Optimizer 的 param_groups 中添加一组参数。

该方法可以在微调(fine-tuning)预训练网络时发挥作用，在训练过程中，冻结层通过此方法加入到 Optimizer 中变为可训练层。

参数

param_group (dict) – 指定了应与参数组一起进行优化的张量。

上一页

megengine.optimizer.AdamW

下一页

megengine.optimizer.AdamW.backward