fix issue#24 #24

Signed-off-by: Ranganath Krishnan <[email protected]>

Author: ranganathkrishnan

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