Skip to content

Commit

Permalink
rename
Browse files Browse the repository at this point in the history
  • Loading branch information
iProzd committed Dec 18, 2024
1 parent 4fdc64f commit d57b97b
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 8 deletions.
4 changes: 2 additions & 2 deletions deepmd/dpmodel/descriptor/dpa3.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ def __init__(
update_style: str = "res_residual",
update_residual: float = 0.1,
update_residual_init: str = "const",
skip_static: bool = False,
skip_stat: bool = False,
) -> None:
r"""The constructor for the RepFlowArgs class which defines the parameters of the repflow block in DPA3 descriptor.
Expand Down Expand Up @@ -77,7 +77,7 @@ def __init__(
self.update_style = update_style
self.update_residual = update_residual
self.update_residual_init = update_residual_init
self.skip_static = skip_static
self.skip_stat = skip_stat

def __getitem__(self, key):
if hasattr(self, key):
Expand Down
2 changes: 1 addition & 1 deletion deepmd/pt/model/descriptor/dpa3.py
Original file line number Diff line number Diff line change
Expand Up @@ -157,7 +157,7 @@ def init_subclass_params(sub_data, sub_class):
update_style=self.repflow_args.update_style,
update_residual=self.repflow_args.update_residual,
update_residual_init=self.repflow_args.update_residual_init,
skip_static=self.repflow_args.skip_static,
skip_stat=self.repflow_args.skip_stat,
exclude_types=exclude_types,
env_protection=env_protection,
precision=precision,
Expand Down
8 changes: 4 additions & 4 deletions deepmd/pt/model/descriptor/repflows.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ def __init__(
exclude_types: list[tuple[int, int]] = [],
env_protection: float = 0.0,
precision: str = "float64",
skip_static: bool = True,
skip_stat: bool = True,
seed: Optional[Union[int, list[int]]] = None,
) -> None:
r"""
Expand Down Expand Up @@ -177,7 +177,7 @@ def __init__(
self.split_sel = self.sel
self.axis_neuron = axis_neuron
self.set_davg_zero = set_davg_zero
self.skip_static = skip_static
self.skip_stat = skip_stat

self.n_dim = n_dim
self.e_dim = e_dim
Expand Down Expand Up @@ -233,7 +233,7 @@ def __init__(
wanted_shape = (self.ntypes, self.nnei, 4)
mean = torch.zeros(wanted_shape, dtype=self.prec, device=env.DEVICE)
stddev = torch.ones(wanted_shape, dtype=self.prec, device=env.DEVICE)
if self.skip_static:
if self.skip_stat:
stddev = stddev * 0.3
self.register_buffer("mean", mean)
self.register_buffer("stddev", stddev)
Expand Down Expand Up @@ -525,7 +525,7 @@ def compute_input_stats(
The path to the stat file.
"""
if self.skip_static and self.set_davg_zero:
if self.skip_stat and self.set_davg_zero:
return
env_mat_stat = EnvMatStatSe(self)
if path is not None:
Expand Down
2 changes: 1 addition & 1 deletion deepmd/utils/argcheck.py
Original file line number Diff line number Diff line change
Expand Up @@ -1483,7 +1483,7 @@ def dpa3_repflow_args():
doc=doc_axis_neuron,
),
Argument(
"skip_static",
"skip_stat",
bool,
optional=True,
default=False,
Expand Down

0 comments on commit d57b97b

Please sign in to comment.