torch.optim.Optimizer.add_param_group

Optimizer.add_param_group(param_group)

將一個參數群組添加到 Optimizer 的 param_groups。

當對預先訓練的網路進行微調時，這可能很有用，因為可以使凍結的層可訓練並在訓練過程中將其添加到 Optimizer。

參數

param_group (dict) – 指定應與群組特定的優化選項一起優化的 Tensor。