xnn.common.finetune.freeze

Freezing parameters by name pattern (layer-freezing fine-tuning).

Functions

count_parameters(model[, trainable_only])	Number of parameters of model (optionally only the trainable ones).
freeze_parameters(model[, freeze, train_only])	Set requires_grad of the parameters of model by name pattern.