Index

B

BayesBatchNorm2d (class in torchbnn.modules.batchnorm) BayesConv2d (class in torchbnn.modules.conv) bayesian_kl_loss() (in module torchbnn.functional)

BayesLinear (class in torchbnn.modules.linear) BayesModule (class in torchbnn.modules.module) BKLLoss (class in torchbnn.modules.loss)

E

extra_repr() (torchbnn.modules.linear.BayesLinear method)

F

forward() (torchbnn.modules.conv.BayesConv2d method) (torchbnn.modules.linear.BayesLinear method) (torchbnn.modules.loss.BKLLoss method)

freeze() (in module torchbnn.utils.freeze_model) (torchbnn.modules.module.BayesModule method)

T

torchbnn.functional (module) torchbnn.modules.batchnorm (module) torchbnn.modules.conv (module)

torchbnn.modules.linear (module) torchbnn.modules.loss (module) torchbnn.modules.module (module) torchbnn.utils.freeze_model (module)

U

unfreeze() (in module torchbnn.utils.freeze_model) (torchbnn.modules.module.BayesModule method)