From e402089556e7e3135100f5ee609f9d0c1ecfa2f3 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 2 May 2022 00:11:26 -0600 Subject: [PATCH] abstractify --- codes/scripts/validate_data.py | 2 +- codes/train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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))