From b8577546264a3aeee025f2763dd63fde44c9dea0 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Wed, 2 Nov 2022 18:00:05 +0530
Subject: [PATCH] add documentation

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

diff --git a/enhancer/models/model.py b/enhancer/models/model.py
index 92d30ae..c679669 100644
--- a/enhancer/models/model.py
+++ b/enhancer/models/model.py
@@ -37,7 +37,7 @@ class Model(pl.LightningModule):
             Enhancer dataset used for training/validation
         duration: float, optional
             duration used for training/inference
-        loss : string or List of strings, default to "mse"
+        loss : string or List of strings or custom loss (nn.Module), default to "mse"
             loss functions to be used. Available ("mse","mae","Si-SDR")
 
     """