diff --git a/modules/util/viirs_l1b_l2.py b/modules/util/viirs_l1b_l2.py
index 6cbdf3155af1e47ad0e27698d3b65b9242244c9c..dec7098f0e19fea0b7272fa196af434f79b20e59 100644
--- a/modules/util/viirs_l1b_l2.py
+++ b/modules/util/viirs_l1b_l2.py
@@ -31,6 +31,9 @@ def run_all(directory, out_directory):
     num_train_samples, num_valid_samples = 0, 0
 
     cnt = 10
+    total_num_train_samples = 0
+    total_num_test_samples = 0
+
     for p in os.scandir(directory):
         if not p.is_dir():
             continue
@@ -143,6 +146,10 @@ def run_all(directory, out_directory):
         num_valid_samples += data_valid.shape[0]
 
         print('num_train_samples, num_valid_samples: ', num_train_samples, num_valid_samples)
+        total_num_train_samples += num_train_samples
+        total_num_test_samples += num_valid_samples
+
+    print('total_num_train_samples, total_num_valid_samples: ', num_train_samples, num_valid_samples)
 
 
 def run(data_h5f, label_h5f, data_tiles, label_tiles, mod_tile_width=64, kernel_size=9):