Skip to content

docs: clarify PyTorch compression support - #401

Open
njzjz-bot wants to merge 1 commit into
deepmodeling:masterfrom
njzjz-bot:docs/issue-274-pytorch-compression
Open

docs: clarify PyTorch compression support#401
njzjz-bot wants to merge 1 commit into
deepmodeling:masterfrom
njzjz-bot:docs/issue-274-pytorch-compression

Conversation

@njzjz-bot

Copy link
Copy Markdown

Summary

  • explain that TensorFlow and PyTorch model files are not interchangeable
  • identify why plain dp compress parses a PyTorch model as a GraphDef
  • document that DeePMD-kit 3.0.0b3 has no PyTorch compression command
  • recommend uncompressed PyTorch deployment or a supported backend/version
  • make the training-backend schema description version-aware

Tests

  • python -m py_compile dpgen2/op/run_dp_train.py
  • isort --check-only dpgen2/op/run_dp_train.py
  • git diff --check

Closes #274

Coding agent: Codex
Codex version: codex-cli 0.149.0
Model: gpt-5.6-sol
Reasoning effort: xhigh

@dosubot dosubot Bot added size:L This PR changes 100-499 lines, ignoring generated files. documentation Improvements or additions to documentation labels Aug 23, 2026
@coderabbitai

coderabbitai Bot commented Aug 23, 2026

Copy link
Copy Markdown

Warning

Review limit reached

Next included review available in 59 minutes.

View limit details

Limit details: You’ve used the included review currently available.

You've used all free OSS reviews for now. Wait for the free limit to reset to keep reviewing this public repository.

Learn how review limits work.

Review configuration:

⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro Plus

Run ID: c488d772-5853-437f-bccf-8144001c1988

📥 Commits

Reviewing files that changed from the base of the PR and between 6b01f29 and b8e1c39.

📒 Files selected for processing (2)
  • docs/input.md
  • dpgen2/op/run_dp_train.py

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands.

Explain backend-specific model formats and the DeePMD-kit 3.0.0b3 limitation that leaves PyTorch compression unavailable.

Coding-Agent: Codex
Codex-Version: codex-cli 0.149.1
Model: gpt-5.6-sol
Reasoning-Effort: xhigh
@njzjz-bot
njzjz-bot force-pushed the docs/issue-274-pytorch-compression branch from 3235a5a to b8e1c39 Compare August 26, 2026 11:00
@dosubot dosubot Bot added size:S This PR changes 10-29 lines, ignoring generated files. and removed size:L This PR changes 100-499 lines, ignoring generated files. labels Aug 26, 2026
@codecov

codecov Bot commented Aug 26, 2026

Copy link
Copy Markdown

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 84.43%. Comparing base (6b01f29) to head (b8e1c39).

Additional details and impacted files
@@           Coverage Diff           @@
##           master     #401   +/-   ##
=======================================
  Coverage   84.43%   84.43%           
=======================================
  Files         104      104           
  Lines        6110     6110           
=======================================
  Hits         5159     5159           
  Misses        951      951           

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

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

Labels

documentation Improvements or additions to documentation size:S This PR changes 10-29 lines, ignoring generated files.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

The dp compress with pytorch backend can not do the compression (finetune-->distillation based on deepmd-kit-v3-b3, dpgen2)

1 participant