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