Upload MPHGNet v4.1 model + weights
Browse files- README.md +36 -0
- model_weights.pt +3 -0
- mphgnet_v41.py +0 -0
README.md
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
tags:
|
| 4 |
+
- physics
|
| 5 |
+
- materials-science
|
| 6 |
+
- graph-neural-network
|
| 7 |
+
- equivariant
|
| 8 |
+
license: mit
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# MPHGNet v4.1
|
| 12 |
+
|
| 13 |
+
Multi-Physics Hierarchical Graph Network (MPHGNet) v4.1 for grain-boundary energy,
|
| 14 |
+
forces, stress and multi-fidelity materials property prediction.
|
| 15 |
+
|
| 16 |
+
## Model Description
|
| 17 |
+
- Equivariant physics backbone with e3nn irreps
|
| 18 |
+
- Multi-physics channels: local density, gradient flow, interaction energy,
|
| 19 |
+
global context, structural stability, Hilbert-space coherence, defect, diffusion, phonon
|
| 20 |
+
- Hebbian STDP plasticity module
|
| 21 |
+
- Triplet + quadruplet physics message passing
|
| 22 |
+
|
| 23 |
+
## Usage
|
| 24 |
+
```python
|
| 25 |
+
from huggingface_hub import hf_hub_download
|
| 26 |
+
import torch, importlib.util, sys
|
| 27 |
+
|
| 28 |
+
path = hf_hub_download(repo_id="harinag0204/MPHGNet-v41", filename="mphgnet_v41.py")
|
| 29 |
+
spec = importlib.util.spec_from_file_location("mphgnet_v41", path)
|
| 30 |
+
mph = importlib.util.module_from_spec(spec); spec.loader.exec_module(mph)
|
| 31 |
+
|
| 32 |
+
model = mph.EquivariantPhysicsBackboneV41(num_layers=1).to(mph.device)
|
| 33 |
+
weights = hf_hub_download(repo_id="harinag0204/MPHGNet-v41", filename="model_weights.pt")
|
| 34 |
+
model.load_state_dict(torch.load(weights, map_location=mph.device))
|
| 35 |
+
model.eval()
|
| 36 |
+
```
|
model_weights.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:edc2ce0ce7ba0e6994c31b50ea1ceaf2e70430a45d0ee7c7985a7c13f3e553d3
|
| 3 |
+
size 21275335
|
mphgnet_v41.py
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|