fix issue#24 #24

Signed-off-by: Ranganath Krishnan <[email protected]>
IntelLabs · Jan 2, 2024 · 1180b87 · 1180b87
1 parent 6677599
commit 1180b87
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/bayesian_torch/layers/flipout_layers/conv_flipout.py b/bayesian_torch/layers/flipout_layers/conv_flipout.py
@@ -238,7 +238,6 @@ def forward(self, x, return_kl=True):
             eps_kernel = self.qint_quant[2](eps_kernel) # random variable
             delta_kernel =self.qint_quant[3](delta_kernel) # multiply activation
 
-        self.kl = kl
         # returning outputs + perturbations
         if return_kl:
             return outputs + perturbed_outputs, kl
@@ -632,7 +631,6 @@ def forward(self, x, return_kl=True):
             eps_kernel = self.qint_quant[2](eps_kernel) # random variable
             delta_kernel =self.qint_quant[3](delta_kernel) # multiply activation
 
-        self.kl = kl
         # returning outputs + perturbations
         if return_kl:
             return outputs + perturbed_outputs, kl
@@ -827,7 +825,6 @@ def forward(self, x, return_kl=True):
             eps_kernel = self.qint_quant[2](eps_kernel) # random variable
             delta_kernel =self.qint_quant[3](delta_kernel) # multiply activation
 
-        self.kl = kl
         # returning outputs + perturbations
         if return_kl:
             return outputs + perturbed_outputs, kl
@@ -1027,7 +1024,6 @@ def forward(self, x, return_kl=True):
             eps_kernel = self.qint_quant[2](eps_kernel) # random variable
             delta_kernel =self.qint_quant[3](delta_kernel) # multiply activation
 
-        self.kl = kl
         # returning outputs + perturbations
         if return_kl:
             return outputs + perturbed_outputs, kl
@@ -1226,7 +1222,6 @@ def forward(self, x, return_kl=True):
             eps_kernel = self.qint_quant[2](eps_kernel) # random variable
             delta_kernel =self.qint_quant[3](delta_kernel) # multiply activation
 
-        self.kl = kl
         # returning outputs + perturbations
         if return_kl:
             return outputs + perturbed_outputs, kl