Solving deprecation warning

FedericoDiMarzo · Mar 1, 2024 · 51c342a · 51c342a
1 parent 55c80f9
commit 51c342a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_utilities/modules.py b/torch_utilities/modules.py
@@ -1,5 +1,5 @@
+from torch.nn.utils.parametrizations import weight_norm
 from typing import Callable, List, Optional, Tuple
-from torch.nn.utils import weight_norm
 import torch.nn.functional as F
 from torch import nn, Tensor
 from torch.nn import Module