yx21e's picture
Refresh WildFIRE-FM model release
84b67b3 verified
# WildFIRE-FM Checkpoints
This directory contains metadata for the released WildFIRE-FM seeded checkpoints.
The checkpoint files are stored on the Hugging Face Hub under:
```text
models/wildfire_fm/checkpoints/seed_1/best_firms_prauc.pt
models/wildfire_fm/checkpoints/seed_7/best_firms_prauc.pt
models/wildfire_fm/checkpoints/seed_42/best_firms_prauc.pt
models/wildfire_fm/checkpoints/seed_99/best_firms_prauc.pt
models/wildfire_fm/checkpoints/seed_123/best_firms_prauc.pt
```
Use `modeling_unet.py` to instantiate the compact U-Net architecture before loading a checkpoint state dict.