GeoGraph / README.md
jeanq1's picture
Update README.md
45bc5b2 verified
|
raw
history blame
327 Bytes
---
license: mit
---
Model can be reloaded with the following code snippet:
import torch
from huggingface_hub import hf_hub_download
p = hf_hub_download("jeanq1/experiment_pel", filename="model.ckpt", local_dir="./weights")
ckpt = torch.load(p, map_location="cpu", weights_only=False) # use it like any local file (modifié)