models/wildfire_fm/README.md

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:

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.