【MIIT program】Add GMTNet for dielectric tensor prediction - #295
【MIIT program】Add GMTNet for dielectric tensor prediction#295cjywahaha-oss wants to merge 6 commits into
Conversation
|
辛苦参考 #258 和套件内已有代码和实现的规范,并提交数据和权重 给出精度对齐结果 |
|
参考已有规范:#258 |
|
Update on GMTNet Paddle reproduction validation: I have pushed a new commit Final local validation on the full test split:
This matches the PyTorch reference result:
Additional checks completed:
Note: The full dielectric dataset is not included in this PR. The full validation was performed locally using the available preprocessed data. |
build_molecule md17 dataset need to rebuild
|
辛苦人工确认 |
Integrate GMTNet model and data support, public prediction paths, precision-preserving CIF parsing, and atomic CIF directory validation.
|
The GMTNet integration has been updated in commit 2767499. The implementation, PropertyPredictor integration, CIF handling, packaging, clean installation, and final CPU regression have been completed. The PR description has also been updated to reflect the final implementation and validation results. Please review it when convenient. Thank you. |
Summary
This PR adds GMTNet support for dielectric tensor prediction in PaddleMaterials.
Main changes
ppmat/models/gmtnet/PropertyPredictorsupport for:pymatgen.Structureinputsprimitive=Falsesort=Falsemerge_tol=0.0frac_tolerance=0.0paddle_scatterimports for models that require it, so GMTNet can be imported and installed independentlyValidation
The following validations were completed:
site-packagesNo checkpoint, training dataset, generated logs, or local
paddle_scattersources are included in the package.Known limitations
Related issue: #194