Skip to content

Commit

Permalink
Remove unnecessary reset_parameters in ExpNormFunction class
Browse files Browse the repository at this point in the history
Signed-off-by: Tsz Wai Ko <[email protected]>
  • Loading branch information
kenko911 committed Sep 23, 2024
1 parent 3bc8467 commit e5ebebc
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions src/matgl/layers/_basis.py
Original file line number Diff line number Diff line change
Expand Up @@ -398,12 +398,6 @@ def _initial_params(self):
betas = torch.tensor([(2 / self.num_rbf * (1 - start_value)) ** -2] * self.num_rbf)
return means, betas

def reset_parameters(self):
"""Reset the means and betas to their initial values."""
means, betas = self._initial_params()
self.means.data.copy_(means)
self.betas.data.copy_(betas)

def forward(self, r: torch.Tensor):
"""
Compute the radial basis function for the input distances.
Expand Down

0 comments on commit e5ebebc

Please sign in to comment.