Update training_multiSample.py

hansenlab · Nov 14, 2024 · 5d75626 · 5d75626
1 parent 70a8d03
commit 5d75626
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]