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

Merge branch 'dev' of https://github.com/shahules786/enhancer into dev-hawk

This commit is contained in:
shahules786 2022-10-06 10:19:38 +05:30
parent 741fd7b87c 3083e1ad7a
commit e06ba07889
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -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!
* A command line tool that facilitates training of highly customisable speech enhacement models from the terminal itself!

10
enhancer/cli/train.py
View File

@ -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__":