update andrew examples

GriffithsLab · Feb 6, 2024 · 72ff8f4 · 72ff8f4
1 parent e60a5b2
commit 72ff8f4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/examples/eg002r__multimodal_simulation.py b/examples/eg002r__multimodal_simulation.py
@@ -80,7 +80,7 @@
 paramsBOLD = BOLD_Params()
 paramsBOLD.to(device)
 
-paramsNode.J = par((0.15  * np.ones(num_regions)), fit_par = True, asLog = True) #This is a parameter that will be updated during training
+paramsNode.J = par((0.15  * np.ones(num_regions)), (0.15  * np.ones(num_regions)), (0.01  * np.ones(num_regions)), fit_par = True, asLog = True) #This is a parameter that will be updated during training
 paramsNode.to(device)
 
 # %%
@@ -155,7 +155,7 @@ def loss(self, simData, empData = None, returnLossComponents = False):
         if returnLossComponents:
             return totalLoss, (S_E_mean_loss.item(), S_I_mean_loss.item(), EEG_PSD_loss.item(), EEG_FC_loss.item(), BOLD_PSD_loss.item(), BOLD_FC_loss.item())
         else:
-            return totalLoss
+            return totalLoss, totalLoss
 
 ObjFun = mmObjectiveFunction()
 

diff --git a/examples/eg005r__gpu_support.py b/examples/eg005r__gpu_support.py
@@ -72,9 +72,9 @@
 # Create a RWW Params
 paramsNode = ParamsRWWEI2(num_regions)
 
-paramsNode.J = par((0.15 * np.ones(num_regions)), fit_par = True, asLog = True) #This is a parameter that will be updated during training
-paramsNode.G = par(torch.tensor(1.0), None, None, True, False, False)
-paramsNode.sig = par(torch.tensor(0.01), None, None, True, False, False)
+paramsNode.J = par((0.15 * np.ones(num_regions)),0.15 * np.ones(num_regions),0.01 * np.ones(num_regions), fit_par = True, asLog = True) #This is a parameter that will be updated during training
+paramsNode.G = par(1,1,0.1, True)
+paramsNode.sig = par(np.log(0.01),np.log(0.01), 0.1, True, True)
 paramsNode.to(device)
 
 #Create #EEG Params