From d94bcde37247adf8557758b50a29b9c665f3dcd1 Mon Sep 17 00:00:00 2001
From: Yi Wang <37149810+yiwang12@users.noreply.github.com>
Date: Thu, 14 Nov 2024 21:21:37 +0100
Subject: [PATCH] Update training_multiSample.py

---
 mNSF/training_multiSample.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mNSF/training_multiSample.py b/mNSF/training_multiSample.py
index d33c16b..087dafb 100644
--- 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"]