Update training_multiSample.py

hansenlab · Nov 14, 2024 · cec07b8 · cec07b8
1 parent f0ca211
commit cec07b8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mNSF/training_multiSample.py b/mNSF/training_multiSample.py
@@ -443,6 +443,8 @@ def _train_model_fixed_lr(self, list_tro,list_Dtrain, list_D__, ckpt_mgr,Dval=No
       	print("vec_batch")
       	print(vec_batch)
       	for ksample in range(0,nsample_chunked):
+		print("ksample")
+		print(ksample)
         	list_tro[ksample].model.Z=list_D__[ksample]["Z"]
         	Dtrain_ksample = list_Dtrain[ksample]
         	if vec_batch[ksample]: