Fix printing total data size

cgrambow · Dec 11, 2019 · 935125a · 935125a
1 parent 18d0a93
commit 935125a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chemprop/train/run_training.py b/chemprop/train/run_training.py
@@ -120,7 +120,7 @@ def run_training(args: Namespace, logger: Logger = None) -> List[float]:
 
     args.train_data_size = len(train_data)
 
-    debug(f'Total size = {len(data):,} | '
+    debug(f'Total size = {len(train_data) + len(val_data) + len(test_data):,} | '
           f'train size = {len(train_data):,} | val size = {len(val_data):,} | test size = {len(test_data):,}')
 
     # Initialize scaler and scale training targets by subtracting mean and dividing standard deviation (regression only)