Fix deprecated usage of torch.nn.utils.weight_norm

SaeedForoutan · web-flow · commit 42ef14a1e3fa · 2024-05-18T06:24:49.000+03:30
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.
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:

Original file line number	Diff line number	Diff line change
`@@ -98,8 +98,8 @@ def __init__(`
`98`	`98`	`)`
`99`	`99`	`if weight_norm:`
`100`	`100`	`self.conv1, self.conv2 = (`
`101`		`- nn.utils.weight_norm(self.conv1),`
`102`		`- nn.utils.weight_norm(self.conv2),`
	`101`	`+ nn.utils.parametrizations.weight_norm(self.conv1),`
	`102`	`+ nn.utils.parametrizations.weight_norm(self.conv2),`
`103`	`103`	`)`
`104`	`104`
`105`	`105`	`if input_dim != output_dim:`