feat(pt): support mixed-nloc LMDB batching - #5439

Closed

littlepeachs wants to merge 20 commits into

deepmodeling:masterfrom

littlepeachs:mix_batch_0429

Commits on Apr 29, 2026

mix batch
liwentao
committed
45bdc59
mix batch revise1
liwentao
committed
effb1a2

Commits on May 6, 2026

lmdb revise 0506
liwentao
committed
83aa004

Commits on May 9, 2026

deleted logs
liwentao
committed
9f4ea0a

Commits on May 18, 2026

Merge branch 'master' into mix_batch_0429
littlepeachs
authored
38c86d6

Commits on May 20, 2026

Add mixed-batch LMDB training support for DPA3
liwentao
committed
57e3ffd
test(lmdb): add local mixed-batch run config
liwentao
committed
f8464f9
Merge branch 'master' into mix_batch_0429
littlepeachs
authored
57369be
fix(pt): restore distributed and lmdb collate imports
liwentao
committed
c30a1d7
fix(pt): guard mixed-batch generalized force loss
liwentao
committed
1656418
style: apply pre-commit formatting
liwentao
committed
e4e1333

Commits on May 21, 2026

fix(pt): skip mixed-batch flat path in torchscript
liwentao
committed
31c24e3
fix(pt): use fixed nlist ghost return type
liwentao
committed
89c226f

Commits on May 23, 2026

feat(pt): add mixed-batch dpmodel coverage
liwentao
committed
8df31f2
chore: remove local mixed-batch debug files
liwentao
committed
f35731c
chore: restore gitignore baseline
liwentao
committed
23c2633

Commits on Jun 26, 2026

feat(pt): refine mixed-nloc LMDB batching
liwentao
committed
67ab717
Merge branch 'master' into mix_batch_0429
liwentao
committed
8230796
fix(pt): align mixed-batch atomic forward hooks
liwentao
committed
8bb2559
fix(pt): preserve energy forward charge-spin ABI
liwentao
committed
c84f9e5