Fix deprecated usage of torch.nn.utils.weight_norm

The previous implementation in darts.darts.models.forecasting.tcn_mode was using `torch.nn.utils.weight_norm`, which is deprecated in favor of `torch.nn.utils.parametrizations.weight_norm`. This commit replaces two occurrences of `torch.nn.utils.weight_norm` with the recommended `torch.nn.utils.parametrizations.weight_norm` to resolve the deprecation warning.
unit8co · May 18, 2024 · 42ef14a · 42ef14a
1 parent 000d29d
commit 42ef14a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/darts/models/forecasting/tcn_model.py b/darts/models/forecasting/tcn_model.py
@@ -98,8 +98,8 @@ def __init__(
         )
         if weight_norm:
             self.conv1, self.conv2 = (
-                nn.utils.weight_norm(self.conv1),
-                nn.utils.weight_norm(self.conv2),
+                nn.utils.parametrizations.weight_norm(self.conv1),
+                nn.utils.parametrizations.weight_norm(self.conv2),
             )
 
         if input_dim != output_dim: