typo fixed in adaptive reg

pollytur · web-flow · commit ad6f591945a1 · 2025-02-27T11:08:41.000+01:00
diff --git a/neuralpredictors/layers/readouts/gaussian.py b/neuralpredictors/layers/readouts/gaussian.py
@@ -399,7 +399,7 @@ def adaptive_feature_l1_lognorm(self, reduction="sum", average=None):
             )
             # adaptive_neuron_reg_coefs (betas) are supposted to be from lognorm distribution
             coef_prior = 1 / (self.gamma_sigma**2) * ((torch.log(self.adaptive_neuron_reg_coefs.abs()) ** 2).sum())
-            return regularization_loss + coef_prior
+            return features_regularization + coef_prior
         else:
             return 0
 

Original file line number	Diff line number	Diff line change
`@@ -399,7 +399,7 @@ def adaptive_feature_l1_lognorm(self, reduction="sum", average=None):`
`399`	`399`	`)`
`400`	`400`	`# adaptive_neuron_reg_coefs (betas) are supposted to be from lognorm distribution`
`401`	`401`	`coef_prior = 1 / (self.gamma_sigma*2) ((torch.log(self.adaptive_neuron_reg_coefs.abs()) ** 2).sum())`
`402`		`- return regularization_loss + coef_prior`
	`402`	`+ return features_regularization + coef_prior`
`403`	`403`	`else:`
`404`	`404`	`return 0`
`405`	`405`