From b071bb171d1caf5e986d25d07292230e4dc51748 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 6 Oct 2022 10:18:31 +0530 Subject: [PATCH] log num steps --- enhancer/cli/train.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/enhancer/cli/train.py b/enhancer/cli/train.py index 5ad61b1..a9c66e0 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,11 +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) - + 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__":