|
|
--- |
|
|
license: mit |
|
|
--- |
|
|
|
|
|
# experiment_pel |
|
|
|
|
|
Pytorch checkpoint for the [GeoGraph](https://github.com/instadeepai/GeoGraph) model (raw PyTorch/Lightning-style checkpoint). |
|
|
|
|
|
## Usage |
|
|
|
|
|
```python |
|
|
import torch |
|
|
from huggingface_hub import hf_hub_download |
|
|
|
|
|
# Download the checkpoint into a local folder (or use the default cache) |
|
|
p = hf_hub_download( |
|
|
repo_id="jeanq1/GeoGraph", |
|
|
filename="model.ckpt", |
|
|
local_dir="./weights" |
|
|
) |
|
|
|
|
|
# Load like any local PyTorch file |
|
|
ckpt = torch.load(p, map_location="cpu", weights_only=False) |
|
|
|
