torch.optim.Optimizer.add_param_group

Optimizer.add_param_group(param_group) [source]

Добавить группу параметров к Optimizer.

Это может быть полезно при дообучении предварительно обученной сети, так как замороженные слои могут быть сделаны обучаемыми и добавлены к Optimizer по мере обучения.

Параметры:

param_group (dict) – Указывает, какие тензоры должны быть оптимизированы вместе с параметрами оптимизации для группы.