評價此頁

torch.optim.Optimizer.add_param_group#

Optimizer.add_param_group(param_group)[來源]#

Optimizerparam_groups 新增一個引數組。

這在微調預訓練網路時可能很有用,因為隨著訓練的進行,可以使凍結的層變得可訓練並新增到 Optimizer 中。

引數

param_group (dict) – 指定哪些 Tensor 應該被最佳化,以及組特定的最佳化選項。