Skip to content

Commit

Permalink
fix(tf): pass type_one_side & exclude_types to DPTabulate in se_r
Browse files Browse the repository at this point in the history
Fix deepmodeling#4445.

* Modify `DPTabulate` instance creation to include `self.type_one_side` and `self.exclude_types`
  • Loading branch information
njzjz committed Nov 30, 2024
1 parent 03c6e49 commit 6bb3133
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions deepmd/tf/descriptor/se_r.py
Original file line number Diff line number Diff line change
Expand Up @@ -356,6 +356,8 @@ def enable_compression(
self.filter_neuron,
graph,
graph_def,
type_one_side=self.type_one_side,
exclude_types=self.exclude_types,
activation_fn=self.filter_activation_fn,
suffix=suffix,
)
Expand Down

0 comments on commit 6bb3133

Please sign in to comment.