From 48a58dc8deade9fa4faa7accb997f174ec618b46 Mon Sep 17 00:00:00 2001 From: Parsa Oveisi Date: Fri, 7 Jun 2024 22:19:45 +0000 Subject: [PATCH] fixed LM prior settings --- whobpyt/datatypes/AbstractNMM.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/whobpyt/datatypes/AbstractNMM.py b/whobpyt/datatypes/AbstractNMM.py index 1e667e5..d0347b0 100644 --- a/whobpyt/datatypes/AbstractNMM.py +++ b/whobpyt/datatypes/AbstractNMM.py @@ -42,7 +42,7 @@ def setModelParameters(self): if (var.fit_par): if var_name == 'lm': size = var.val.shape - var.val = Parameter(- 1 * torch.ones((size[0], size[1])) + 0.1*torch.randn(size[0], size[1])) + var.val = Parameter(var.val) var.prior_mean = Parameter(var.prior_mean) var.prior_precision = Parameter(var.prior_precision) param_reg.append(var.val)