deepmd-kit
deepmd-kit copied to clipboard
Feat: add model format for dpa1
This PR add model format for DPA1 model:
- [x] Add torch reformat implementation for DPA1 model
- [x] Add numpy implementation for DPA1 model without attention layer
- [x] Align the torch and numpy implementations
TODO:
- [ ] Add numpy implementation for DPA1 model with attention layer
- [ ] Align the TF and numpy implementations
- [ ] Align the smoothness implementations
- [ ] Make filter_layers._networks in torch be accessable from outside
Codecov Report
Attention: 529 lines in your changes are missing coverage. Please review.
Comparison is base (
afb440a) 74.39% compared to head (a96cab0) 20.72%. Report is 2 commits behind head on devel.
Additional details and impacted files
@@ Coverage Diff @@
## devel #3211 +/- ##
===========================================
- Coverage 74.39% 20.72% -53.68%
===========================================
Files 345 346 +1
Lines 31981 32509 +528
Branches 1592 1594 +2
===========================================
- Hits 23791 6736 -17055
- Misses 7265 25075 +17810
+ Partials 925 698 -227
:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.
This PR is merged into https://github.com/deepmodeling/deepmd-kit/pull/3696