Update training_multiSample.py

hansenlab · Nov 14, 2024 · d94bcde · d94bcde
1 parent 64b78f4
commit d94bcde
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mNSF/training_multiSample.py b/mNSF/training_multiSample.py
@@ -427,6 +427,8 @@ def _train_model_fixed_lr(self, list_tro,list_Dtrain, list_D__, ckpt_mgr,Dval=No
       #chol=(self.epoch % kernel_hp_update_freq==0)
       trl=0.0
       nsample_chunked=len(list_Dtrain)
+      print("list_nchunk")
+      print(list_nchunk)
       if list_nchunk is None:
       	for ksample in range(0,nsample_chunked):
         	list_tro[ksample].model.Z=list_D__[ksample]["Z"]