Get optimizer error fix.

facebookresearch · Sep 30, 2023 · 24b572c · 24b572c
1 parent a2b9675
commit 24b572c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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