From 1c81d629c42dacc1233492af9e267108e09e1bf0 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 6 Oct 2022 10:17:32 +0530 Subject: [PATCH] num steps --- enhancer/cli/train.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/enhancer/cli/train.py b/enhancer/cli/train.py index cb3c7c1..5ad61b1 100644 --- a/enhancer/cli/train.py +++ b/enhancer/cli/train.py @@ -32,7 +32,7 @@ def main(config: DictConfig): loss=parameters.get("loss"), metric=parameters.get("metric"), ) - + direction = model.valid_monitor checkpoint = ModelCheckpoint( dirpath="./model", @@ -79,6 +79,11 @@ 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__":