diff --git a/README.md b/README.md
index 97e0c7e..8ade768 100644
--- a/README.md
+++ b/README.md
@@ -3,4 +3,4 @@ Enhancer is a Pytorch-based opensource toolkit for speech enhancement. It is des
 
 * Various pretrained models nicely integrated with huggingface that users can select and use without any hastle.
 * Ability to train and validation your own custom speech enhancement models with just under 10 lines of code!
-* A command line tool that facilitates training of highly customisable speech enhacement models from the terminal itself!
\ No newline at end of file
+* A command line tool that facilitates training of highly customisable speech enhacement models from the terminal itself!
diff --git a/enhancer/cli/train.py b/enhancer/cli/train.py
index cb3c7c1..a9c66e0 100644
--- a/enhancer/cli/train.py
+++ b/enhancer/cli/train.py
@@ -79,6 +79,16 @@ def main(config: DictConfig):
     )
     if os.path.isfile(saved_location):
         logger.experiment.log_artifact(logger.run_id, saved_location)
+        logger.experiment.log_param(
+            logger.run_id,
+            "num_train_steps_per_epoch",
+            dataset.train__len__() / dataset.batch_size,
+        )
+        logger.experiment.log_param(
+            logger.run_id,
+            "num_valid_steps_per_epoch",
+            dataset.val__len__() / dataset.batch_size,
+        )
 
 
 if __name__ == "__main__":