Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion deepmd/dpmodel/descriptor/hybrid.py
Original file line number Diff line number Diff line change
Expand Up @@ -245,7 +245,7 @@ def enable_compression(
table_stride_2: float = 0.1,
check_frequency: int = -1,
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
2 changes: 1 addition & 1 deletion deepmd/dpmodel/descriptor/make_base_descriptor.py
Original file line number Diff line number Diff line change
Expand Up @@ -162,7 +162,7 @@ def enable_compression(
table_stride_2: float = 0.1,
check_frequency: int = -1,
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
2 changes: 1 addition & 1 deletion deepmd/pd/model/descriptor/dpa1.py
Original file line number Diff line number Diff line change
Expand Up @@ -600,7 +600,7 @@ def enable_compression(
table_stride_2: float = 0.1,
check_frequency: int = -1,
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
2 changes: 1 addition & 1 deletion deepmd/pd/model/descriptor/se_a.py
Original file line number Diff line number Diff line change
Expand Up @@ -264,7 +264,7 @@ def enable_compression(
table_stride_2: float = 0.1,
check_frequency: int = -1,
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
2 changes: 1 addition & 1 deletion deepmd/pt/model/descriptor/dpa1.py
Original file line number Diff line number Diff line change
Expand Up @@ -579,7 +579,7 @@ def enable_compression(
table_stride_2: float = 0.1,
check_frequency: int = -1,
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
2 changes: 1 addition & 1 deletion deepmd/pt/model/descriptor/hybrid.py
Original file line number Diff line number Diff line change
Expand Up @@ -237,7 +237,7 @@ def enable_compression(
table_stride_2: float = 0.1,
check_frequency: int = -1,
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
2 changes: 1 addition & 1 deletion deepmd/pt/model/descriptor/se_a.py
Original file line number Diff line number Diff line change
Expand Up @@ -262,7 +262,7 @@ def enable_compression(
table_stride_2: float = 0.1,
check_frequency: int = -1,
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
2 changes: 1 addition & 1 deletion deepmd/pt/model/descriptor/se_r.py
Original file line number Diff line number Diff line change
Expand Up @@ -364,7 +364,7 @@ def enable_compression(
table_stride_2: float = 0.1,
check_frequency: int = -1,
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
2 changes: 1 addition & 1 deletion deepmd/pt/model/descriptor/se_t.py
Original file line number Diff line number Diff line change
Expand Up @@ -289,7 +289,7 @@ def enable_compression(
table_stride_2: float = 0.1,
check_frequency: int = -1,
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
4 changes: 2 additions & 2 deletions deepmd/tf/descriptor/descriptor.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,7 @@ def compute_input_stats(
input_dict: dict[str, list[np.ndarray]],
**kwargs: Any,
) -> None:
"""Compute the statisitcs (avg and std) of the training data. The input will be
"""Compute the statistics (avg and std) of the training data. The input will be
normalized by the statistics.

Parameters
Expand Down Expand Up @@ -228,7 +228,7 @@ def enable_compression(
check_frequency: int = -1,
suffix: str = "",
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the
training data.

Parameters
Expand Down
16 changes: 8 additions & 8 deletions deepmd/tf/descriptor/hybrid.py
Original file line number Diff line number Diff line change
Expand Up @@ -144,7 +144,7 @@ def compute_input_stats(
real_natoms_vec: list | None = None,
**kwargs: Any,
) -> None:
"""Compute the statisitcs (avg and std) of the training data. The input will be normalized by the statistics.
"""Compute the statistics (avg and std) of the training data. The input will be normalized by the statistics.
Comment thread
wanghan-iapcm marked this conversation as resolved.

Parameters
----------
Expand Down Expand Up @@ -183,22 +183,22 @@ def compute_input_stats(
)

def merge_input_stats(self, stat_dict: dict[str, float]) -> None:
"""Merge the statisitcs computed from compute_input_stats to obtain the self.davg and self.dstd.
"""Merge the statistics computed from compute_input_stats to obtain the self.davg and self.dstd.

Parameters
----------
stat_dict
The dict of statisitcs computed from compute_input_stats, including:
The dict of statistics computed from compute_input_stats, including:
sumr
The sum of radial statisitcs.
The sum of radial statistics.
suma
The sum of relative coord statisitcs.
The sum of relative coord statistics.
sumn
The sum of neighbor numbers.
sumr2
The sum of square of radial statisitcs.
The sum of square of radial statistics.
suma2
The sum of square of relative coord statisitcs.
The sum of square of relative coord statistics.
"""
for ii in self.descrpt_list:
ii.merge_input_stats(stat_dict)
Expand Down Expand Up @@ -315,7 +315,7 @@ def enable_compression(
check_frequency: int = -1,
suffix: str = "",
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the
training data.

Parameters
Expand Down
2 changes: 1 addition & 1 deletion deepmd/tf/descriptor/loc_frame.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,7 +164,7 @@ def compute_input_stats(
input_dict: dict,
**kwargs: Any,
) -> None:
"""Compute the statisitcs (avg and std) of the training data. The input will be normalized by the statistics.
"""Compute the statistics (avg and std) of the training data. The input will be normalized by the statistics.

Parameters
----------
Expand Down
16 changes: 8 additions & 8 deletions deepmd/tf/descriptor/se_a.py
Original file line number Diff line number Diff line change
Expand Up @@ -355,7 +355,7 @@ def compute_input_stats(
input_dict: dict,
**kwargs: Any,
) -> None:
"""Compute the statisitcs (avg and std) of the training data. The input will be normalized by the statistics.
"""Compute the statistics (avg and std) of the training data. The input will be normalized by the statistics.

Parameters
----------
Expand Down Expand Up @@ -401,22 +401,22 @@ def compute_input_stats(
self.merge_input_stats(stat_dict)

def merge_input_stats(self, stat_dict: dict[str, Any]) -> None:
"""Merge the statisitcs computed from compute_input_stats to obtain the self.davg and self.dstd.
"""Merge the statistics computed from compute_input_stats to obtain the self.davg and self.dstd.

Parameters
----------
stat_dict
The dict of statisitcs computed from compute_input_stats, including:
The dict of statistics computed from compute_input_stats, including:
sumr
The sum of radial statisitcs.
The sum of radial statistics.
suma
The sum of relative coord statisitcs.
The sum of relative coord statistics.
sumn
The sum of neighbor numbers.
sumr2
The sum of square of radial statisitcs.
The sum of square of radial statistics.
suma2
The sum of square of relative coord statisitcs.
The sum of square of relative coord statistics.
"""
all_davg = []
all_dstd = []
Expand Down Expand Up @@ -452,7 +452,7 @@ def enable_compression(
check_frequency: int = -1,
suffix: str = "",
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
2 changes: 1 addition & 1 deletion deepmd/tf/descriptor/se_a_ef.py
Original file line number Diff line number Diff line change
Expand Up @@ -176,7 +176,7 @@ def compute_input_stats(
input_dict: dict,
**kwargs: Any,
) -> None:
"""Compute the statisitcs (avg and std) of the training data. The input will be normalized by the statistics.
"""Compute the statistics (avg and std) of the training data. The input will be normalized by the statistics.

Parameters
----------
Expand Down
2 changes: 1 addition & 1 deletion deepmd/tf/descriptor/se_a_mask.py
Original file line number Diff line number Diff line change
Expand Up @@ -234,7 +234,7 @@ def compute_input_stats(
input_dict: dict,
**kwargs: Any,
) -> None:
"""Compute the statisitcs (avg and std) of the training data. The input will be normalized by the statistics.
"""Compute the statistics (avg and std) of the training data. The input will be normalized by the statistics.

Parameters
----------
Expand Down
6 changes: 3 additions & 3 deletions deepmd/tf/descriptor/se_atten.py
Original file line number Diff line number Diff line change
Expand Up @@ -347,7 +347,7 @@ def compute_input_stats(
real_natoms_vec: list | None = None,
**kwargs: Any,
) -> None:
"""Compute the statisitcs (avg and std) of the training data. The input will be normalized by the statistics.
"""Compute the statistics (avg and std) of the training data. The input will be normalized by the statistics.

Parameters
----------
Expand Down Expand Up @@ -432,7 +432,7 @@ def enable_compression(
suffix: str = "",
tebd_suffix: str = "",
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down Expand Up @@ -2362,7 +2362,7 @@ def enable_compression(
suffix: str = "",
tebd_suffix: str = "",
) -> None:
"""Reveive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
12 changes: 6 additions & 6 deletions deepmd/tf/descriptor/se_r.py
Original file line number Diff line number Diff line change
Expand Up @@ -255,7 +255,7 @@ def compute_input_stats(
input_dict: dict,
**kwargs: Any,
) -> None:
"""Compute the statisitcs (avg and std) of the training data. The input will be normalized by the statistics.
"""Compute the statistics (avg and std) of the training data. The input will be normalized by the statistics.

Parameters
----------
Expand Down Expand Up @@ -288,18 +288,18 @@ def compute_input_stats(
self.merge_input_stats(stat_dict)

def merge_input_stats(self, stat_dict: dict[str, Any]) -> None:
"""Merge the statisitcs computed from compute_input_stats to obtain the self.davg and self.dstd.
"""Merge the statistics computed from compute_input_stats to obtain the self.davg and self.dstd.

Parameters
----------
stat_dict
The dict of statisitcs computed from compute_input_stats, including:
The dict of statistics computed from compute_input_stats, including:
sumr
The sum of radial statisitcs.
The sum of radial statistics.
sumn
The sum of neighbor numbers.
sumr2
The sum of square of radial statisitcs.
The sum of square of radial statistics.
"""
all_davg = []
all_dstd = []
Expand Down Expand Up @@ -329,7 +329,7 @@ def enable_compression(
check_frequency: int = -1,
suffix: str = "",
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
16 changes: 8 additions & 8 deletions deepmd/tf/descriptor/se_t.py
Original file line number Diff line number Diff line change
Expand Up @@ -238,7 +238,7 @@ def compute_input_stats(
input_dict: dict,
**kwargs: Any,
) -> None:
"""Compute the statisitcs (avg and std) of the training data. The input will be normalized by the statistics.
"""Compute the statistics (avg and std) of the training data. The input will be normalized by the statistics.

Parameters
----------
Expand Down Expand Up @@ -284,22 +284,22 @@ def compute_input_stats(
self.merge_input_stats(stat_dict)

def merge_input_stats(self, stat_dict: dict[str, Any]) -> None:
"""Merge the statisitcs computed from compute_input_stats to obtain the self.davg and self.dstd.
"""Merge the statistics computed from compute_input_stats to obtain the self.davg and self.dstd.

Parameters
----------
stat_dict
The dict of statisitcs computed from compute_input_stats, including:
The dict of statistics computed from compute_input_stats, including:
sumr
The sum of radial statisitcs.
The sum of radial statistics.
suma
The sum of relative coord statisitcs.
The sum of relative coord statistics.
sumn
The sum of neighbor numbers.
sumr2
The sum of square of radial statisitcs.
The sum of square of radial statistics.
suma2
The sum of square of relative coord statisitcs.
The sum of square of relative coord statistics.
"""
all_davg = []
all_dstd = []
Expand Down Expand Up @@ -335,7 +335,7 @@ def enable_compression(
check_frequency: int = -1,
suffix: str = "",
) -> None:
"""Receive the statisitcs (distance, max_nbor_size and env_mat_range) of the training data.
"""Receive the statistics (distance, max_nbor_size and env_mat_range) of the training data.

Parameters
----------
Expand Down
Loading