nip.utils.training

nip.utils.training#

Utilities to help with training models.

Classes

ParamGroupFreezer(optimizer, ...[, ...])

A class for freezing groups of parameters during training.