Utils¶

Freeze Model¶

torchbnn.utils.freeze_model.freeze(module)[source]¶

Methods for freezing bayesian-model.

Parameters:	model (nn.Module) – a model to be freezed.

torchbnn.utils.freeze_model.unfreeze(module)[source]¶

Methods for unfreezing bayesian-model.

Parameters:	model (nn.Module) – a model to be unfreezed.