Class DenseNetAdjoint

tests/visualize/test_occlusion_sensitivity.py:24–29

class DenseNetAdjoint(DenseNet121):

    def __call__(self, x, adjoint_info):
        if adjoint_info != 42:
            raise ValueError
        return super().__call__(x)


device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")