Use torch.linalg.norm instead of torch.norm (#515)

torch.norm is deprecated and may be removed in a future PyTorch release.
aiqm · Sep 26, 2020 · ca6e325 · ca6e325
1 parent 31bf913
commit ca6e325
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchani/utils.py b/torchani/utils.py
@@ -338,7 +338,7 @@ def vibrational_analysis(masses, hessian, mode_type='MDU', unit='cm^-1'):
     # Note that the normal modes are the COLUMNS of the eigenvectors matrix
     mw_normalized = eigenvectors.t()
     md_unnormalized = mw_normalized * inv_sqrt_mass
-    norm_factors = 1 / torch.norm(md_unnormalized, dim=1)  # units are sqrt(AMU)
+    norm_factors = 1 / torch.linalg.norm(md_unnormalized, dim=1)  # units are sqrt(AMU)
     md_normalized = md_unnormalized * norm_factors.unsqueeze(1)
 
     rmasses = norm_factors**2  # units are AMU