Skip to content

enable model compression for dipole and polar type fitting net#1228

Merged
wanghan-iapcm merged 2 commits intodeepmodeling:develfrom
denghuilu:dipole-compression
Oct 21, 2021
Merged

enable model compression for dipole and polar type fitting net#1228
wanghan-iapcm merged 2 commits intodeepmodeling:develfrom
denghuilu:dipole-compression

Conversation

@denghuilu
Copy link
Member

main changes:

  • add init_variables for dipole and polar fitting net
  • change the parameter of the init_variables function to model_file in the fitting net to match the form of init_variables in descriptors.

This PR should fix issue #1226 .

@codecov-commenter
Copy link

codecov-commenter commented Oct 21, 2021

Codecov Report

Merging #1228 (db5c821) into devel (1cf3e50) will decrease coverage by 0.01%.
The diff coverage is 72.72%.

Impacted file tree graph

@@            Coverage Diff             @@
##            devel    #1228      +/-   ##
==========================================
- Coverage   76.03%   76.02%   -0.02%     
==========================================
  Files          91       91              
  Lines        7358     7367       +9     
==========================================
+ Hits         5595     5601       +6     
- Misses       1763     1766       +3     
Impacted Files Coverage Δ
deepmd/descriptor/se.py 68.42% <ø> (ø)
deepmd/train/trainer.py 73.89% <50.00%> (ø)
deepmd/fit/polar.py 48.74% <70.00%> (+0.55%) ⬆️
deepmd/fit/dipole.py 95.65% <85.71%> (-1.28%) ⬇️
deepmd/fit/ener.py 94.60% <100.00%> (-0.03%) ⬇️

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 1cf3e50...db5c821. Read the comment docs.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[FEATURE_REQUEST] "dp compress" supports only the compression of energy model?

4 participants