`test_bayesian_resnet`¶

Module Contents¶

Classes¶

TestBayesianResNet A suite of tests on ResNet models

class test_bayesian_resnet.TestBayesianResNet(methodName='runTest')[source]¶

Bases: unittest.TestCase

A suite of tests on ResNet models

classmethod setUpClass(cls)[source]¶: Hook method for setting up class fixture before running tests in the class.

test_bayesian_resnet_forward(self)[source]¶: Test instantiation and forward method of BayesianResNet

test_bayesian_resnet_dropout(self)[source]¶: Test if dropout rate is propagated to basic blocks

test_activation_maps_layer3(self)[source]¶: Test if the 3-layer BNN has the correctly shaped activation maps (intermediate feature maps)

test_activation_maps_layer4(self)[source]¶: Test if the 4-layer BNN has the correctly shaped activation maps