From e9d2773b67815ee980384084a1f1a260a190265f Mon Sep 17 00:00:00 2001 From: jrichter <jrichter@exchange.informatik.uni-hamburg.de> Date: Tue, 18 Jun 2024 08:22:39 +0200 Subject: [PATCH] 3e-2 -> 0.03 for clarity --- sgmse/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sgmse/model.py b/sgmse/model.py index a97a566..3c4bf8b 100644 --- a/sgmse/model.py +++ b/sgmse/model.py @@ -18,13 +18,13 @@ class ScoreModel(pl.LightningModule): def add_argparse_args(parser): parser.add_argument("--lr", type=float, default=1e-4, help="The learning rate (1e-4 by default)") parser.add_argument("--ema_decay", type=float, default=0.999, help="The parameter EMA decay constant (0.999 by default)") - parser.add_argument("--t_eps", type=float, default=0.03, help="The minimum time (3e-2 by default)") + parser.add_argument("--t_eps", type=float, default=0.03, help="The minimum process time (0.03 by default)") parser.add_argument("--num_eval_files", type=int, default=20, help="Number of files for speech enhancement performance evaluation during training. Pass 0 to turn off (no checkpoints based on evaluation metrics will be generated).") parser.add_argument("--loss_type", type=str, default="mse", choices=("mse", "mae"), help="The type of loss function to use.") return parser def __init__( - self, backbone, sde, lr=1e-4, ema_decay=0.999, t_eps=3e-2, + self, backbone, sde, lr=1e-4, ema_decay=0.999, t_eps=0.03, num_eval_files=20, loss_type='mse', data_module_cls=None, **kwargs ): """ -- GitLab