Closed kenko911 closed 3 months ago
The updates focus on transitioning from the MEGNet model to the M3GNet model within a PyTorch Lightning framework for materials science applications. Key modifications include renaming the model across various sections, introducing an include_line_graph
parameter to enhance dataset and model configuration, and refining data handling with functools.partial
for improved flexibility. This transition marks a significant update aimed at leveraging M3GNet's capabilities for predicting formation energy and potential.
File Path | Changes |
---|---|
examples/.../Training a M3GNet Formation Energy Model with PyTorch Lightning.ipynb |
- Renamed "MEGNet" to "M3GNet" - Added include_line_graph parameter |
examples/.../Training a M3GNet Potential with PyTorch Lightning.ipynb |
- Added functools.partial import - Included include_line_graph=True in dataset and model configurations |
🐰✨
In the realm of atoms and bonds,
Where data like a river flows,
M3GNet rises, magic wands,
Transforming zeros into pros.With each update, a leap, a bound,
CodeRabbit hops through lines of code,
In PyTorch Lightning, solutions found,
On AI's ever-brightening road.
🌟🐾
Thank you for using CodeRabbit. We offer it for free to the OSS community and would appreciate your support in helping us grow. If you find it useful, would you consider giving us a shout-out on your favorite social media?
Summary
Including include_line_graph=True for M3GNet training in the jupyter notebook
Checklist
ruff
.mypy
.duecredit
@due.dcite
decorators to reference relevant papers by DOI (example)Tip: Install
pre-commit
hooks to auto-check types and linting before every commit:Summary by CodeRabbit
include_line_graph
parameter during dataset conversion and model setup to enhance model configuration flexibility.