diff --git a/codes/scripts/validate_data.py b/codes/scripts/validate_data.py
index e88d4f54..2435386e 100644
--- a/codes/scripts/validate_data.py
+++ b/codes/scripts/validate_data.py
@@ -46,7 +46,7 @@ def main():
             dataset_opt['n_workers'] = 0  # Force num_workers=0 to make dataloader work in process.
             train_loader = create_dataloader(train_set, dataset_opt, opt, None)
             if rank <= 0:
-                print('Number of train images: {:,d}, iters: {:,d}'.format(
+                print('Number of training data elements: {:,d}, iters: {:,d}'.format(
                     len(train_set), train_size))
     assert train_loader is not None
 
diff --git a/codes/train.py b/codes/train.py
index f13e8e55..f40435f5 100644
--- a/codes/train.py
+++ b/codes/train.py
@@ -121,7 +121,7 @@ class Trainer:
                     shuffle = True
                 self.train_loader = create_dataloader(self.train_set, dataset_opt, opt, self.train_sampler, collate_fn=collate_fn, shuffle=shuffle)
                 if self.rank <= 0:
-                    self.logger.info('Number of train images: {:,d}, iters: {:,d}'.format(
+                    self.logger.info('Number of training data elements: {:,d}, iters: {:,d}'.format(
                         len(self.train_set), train_size))
                     self.logger.info('Total epochs needed: {:d} for iters {:,d}'.format(
                         self.total_epochs, total_iters))