dba52ab3214ef3438d27d9642647b6537ae00f7e,tests/get_activations_test.py,GetActivationsTest,test_gradients_of_activations,#GetActivationsTest#,211

Before Change

        self.assertListEqual(list(acts_nested), list(grad_acts_nested))
        self.assertListEqual(list(grad_acts_nested["i1"].shape), list(acts_nested["i1"].shape))
        self.assertListEqual(list(grad_acts_nested["model/fc1"].shape), list(acts_nested["model/fc1"].shape))
        self.assertListEqual(list(grad_acts_nested["model/fc1"].shape), list(acts_nested["model/fc1"].shape))
        self.assertListEqual(list(grad_acts_nested["block/fc1"].shape), list(acts_nested["block/fc1"].shape))
        self.assertListEqual(list(grad_acts_nested["block/relu"].shape), list(acts_nested["block/relu"].shape))
        self.assertListEqual(list(grad_acts_nested["fc1"].shape), list(acts_nested["fc1"].shape))

After Change

        acts_nested = get_activations(model, x, nested=True)

        // same support without the input.
        support_without_input = list(acts)
        support_without_input.remove("i1")
        self.assertListEqual(support_without_input, list(grad_acts))
        self.assertListEqual(list(grad_acts["model"].shape), list(acts["model"].shape))
        self.assertListEqual(list(grad_acts["block"].shape), list(acts["block"].shape))
        self.assertListEqual(list(grad_acts["fc1"].shape), list(acts["fc1"].shape))

Instances