From 5d75626db6da522f7ecc50b13287cc644f5d6df1 Mon Sep 17 00:00:00 2001 From: Yi Wang <37149810+yiwang12@users.noreply.github.com> Date: Thu, 14 Nov 2024 21:06:32 +0100 Subject: [PATCH] Update training_multiSample.py --- mNSF/training_multiSample.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mNSF/training_multiSample.py b/mNSF/training_multiSample.py index 1270136..ddc5178 100644 --- a/mNSF/training_multiSample.py +++ b/mNSF/training_multiSample.py @@ -438,7 +438,7 @@ def _train_model_fixed_lr(self, list_tro,list_Dtrain, list_D__, ckpt_mgr,Dval=No else: vec_batch = list() for ksample in range(0,nsample): - vec_batch.append([False]*1 + [True]*(ist_nchunk[ksample]-1)) + vec_batch.append([False]*1 + [True]*(list_nchunk[ksample]-1)) for ksample in range(0,nsample): list_tro[ksample].model.Z=list_D__[ksample]["Z"] Dtrain_ksample = list_Dtrain[ksample]