From ab0f51ea176c4a1e2e65b5e16b08dfc751be0448 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Mon, 10 Oct 2022 15:33:08 +0530
Subject: [PATCH 1/4] log metric

---
 enhancer/models/model.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index 07564cd..4f1d75b 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -181,11 +181,13 @@ class Model(pl.LightningModule):
             value = metric(target, prediction)
             metric_dict[metric.name] = value
 
-        self.logger.experiment.log_metrics(
-            run_id=self.logger.run_id,
-            metrics=metric_dict,
-            step=self.global_step,
-        )
+        for k, v in metric_dict.items():
+            self.logger.experiment.log_metric(
+                run_id=self.logger.run_id,
+                key=k,
+                value=v,
+                step=self.global_step,
+            )
 
         return metric_dict
 

From a72f3072832853c3f0b92d7d7587f916ece3410b Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Mon, 10 Oct 2022 17:41:58 +0530
Subject: [PATCH 2/4] readme'

---
 README.md | 42 ++++++++++++++++++++++++++++++++++++++----
 1 file changed, 38 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 8ade768..d02e39e 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,40 @@
-# enhancer
-Enhancer is a Pytorch-based opensource toolkit for speech enhancement. It is designed to save time for audio researchers. Is provides easy to use pretrained audio enhancement models and facilitates highly customisable custom model training . Enhancer provides
+# mayavoz
+mayavoz is a Pytorch-based opensource toolkit for speech enhancement. It is designed to save time for audio researchers. Is provides easy to use pretrained audio enhancement models and facilitates highly customisable custom model training .
 
-* 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!
+| **[Quick Start]()** | **[Installation]()** | **[Tutorials]()** | **[Available Recipes]()**
+## Key features :key:
+
+* Various pretrained models nicely integrated with huggingface 	:hugs: that users can select and use without any hastle.
+* :package: 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!
+* Supports multi-gpu training integrated with Pytorch Lightning.
+
+## Quick Start
+``` python
+from mayavoz import Mayamodel
+
+model = Mayamodel.from_pretrained("mayavoz/waveunet")
+model("noisy_audio.wav")
+```
+
+## Installation
+Only Python 3.8+ is officially supported (though it might work with Python 3.7)
+
+- With Pypi
+```
+pip install mayavoz
+```
+
+- With conda
+
+```
+conda env create -f environment.yml
+conda activate mayavoz
+```
+
+- From source code
+```
+git clone url
+cd mayavoz
+pip install -e .
+```

From 4363821c7525ab1382c3d63de6e2b7cbdc4cd5cf Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Mon, 10 Oct 2022 17:45:34 +0530
Subject: [PATCH 3/4] add emojis

---
 README.md | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index d02e39e..f721ab0 100644
--- a/README.md
+++ b/README.md
@@ -6,10 +6,10 @@ mayavoz is a Pytorch-based opensource toolkit for speech enhancement. It is desi
 
 * Various pretrained models nicely integrated with huggingface 	:hugs: that users can select and use without any hastle.
 * :package: 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!
-* Supports multi-gpu training integrated with Pytorch Lightning.
+* :magic_wand: A command line tool that facilitates training of highly customisable speech enhacement models from the terminal itself!
+* :zap: Supports multi-gpu training integrated with Pytorch Lightning.
 
-## Quick Start
+## Quick Start :fire:
 ``` python
 from mayavoz import Mayamodel
 

From c193a48e8e27a2779273bb28d7600ef133fea858 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Mon, 10 Oct 2022 21:03:39 +0530
Subject: [PATCH 4/4] empty cache

---
 enhancer/models/model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index 4f1d75b..7dbc065 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -110,6 +110,7 @@ class Model(pl.LightningModule):
 
     def setup(self, stage: Optional[str] = None):
         if stage == "fit":
+            torch.cuda.empty_cache()
             self.dataset.setup(stage)
             self.dataset.model = self