Update training_multiSample.py

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