From 24b572ccd7a82933305958df47170ba08ababca9 Mon Sep 17 00:00:00 2001 From: Igor Skovorodkin Date: Sun, 1 Oct 2023 00:07:11 +0200 Subject: [PATCH] Get optimizer error fix. --- audiocraft/solvers/builders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/audiocraft/solvers/builders.py b/audiocraft/solvers/builders.py index 304d8f08..7cbb8e5c 100644 --- a/audiocraft/solvers/builders.py +++ b/audiocraft/solvers/builders.py @@ -108,7 +108,7 @@ def get_optimizer(params: tp.Union[nn.Module, tp.Iterable[torch.Tensor]], cfg: o elif cfg.optimizer == 'dadam': optimizer = optim.DAdaptAdam(parameters, lr=cfg.lr, **cfg.adam) else: - raise ValueError(f"Unsupported LR Scheduler: {cfg.lr_scheduler}") + raise ValueError(f"Unsupported Optimizer: {cfg.optimizer}") return optimizer