Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

log num steps

This commit is contained in:
shahules786 2022-10-06 10:18:31 +05:30
parent 1c81d629c4
commit b071bb171d
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
enhancer/cli/train.py
View File

@ -32,7 +32,7 @@ def main(config: DictConfig):
loss=parameters.get("loss"), loss=parameters.get("loss"),
metric=parameters.get("metric"), metric=parameters.get("metric"),
) )
direction = model.valid_monitor direction = model.valid_monitor
checkpoint = ModelCheckpoint( checkpoint = ModelCheckpoint(
dirpath="./model", dirpath="./model",
@ -79,11 +79,16 @@ def main(config: DictConfig):
) )
if os.path.isfile(saved_location): if os.path.isfile(saved_location):
logger.experiment.log_artifact(logger.run_id, saved_location) logger.experiment.log_artifact(logger.run_id, saved_location)
logger.experiment.log_param(logger.run_id, "num_train_steps_per_epoch", logger.experiment.log_param(
dataset.train__len__() / dataset.batch_size) logger.run_id,
logger.experiment.log_param(logger.run_id, "num_valid_steps_per_epoch", "num_train_steps_per_epoch",
dataset.val__len__() / dataset.batch_size) 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__": if __name__ == "__main__":