Skip to content

Commit

Permalink
fix(tf): pass type_one_side & exclude_types to DPTabulate in `s…
Browse files Browse the repository at this point in the history
…e_r` (deepmodeling#4446)

Fix deepmodeling#4445.

* Modify `DPTabulate` instance creation to include `self.type_one_side`
and `self.exclude_types`

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

- **New Features**
- Enhanced configurability for the `DescrptSeR` class, allowing users to
customize compression behavior with new parameters.
- Introduced optional parameters for improved management of atom types
and interactions during the embedding process.
  
- **Bug Fixes**
- Added validation for excluded types to ensure proper handling within
the compression logic.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->
  • Loading branch information
njzjz authored Dec 2, 2024
1 parent db0a2a3 commit 9b70351
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 9b70351

Please sign in to comment.