File size: 523 Bytes
45bc5b2 ffa28f6 ed951f7 ffa28f6 45bc5b2 ffa28f6 f58153b ffa28f6 6c354f1 ffa28f6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
---
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)
|