Add FireWx-FM training and data loader pipeline

README.md

@@ -37,6 +37,7 @@ pretty_name: WildFIRE-FM
 <p>
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM/blob/main/paper/wildfire_fm_evaluation_contracts.pdf"><b>Read the paper PDF</b></a> ·
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM/tree/main/models/wildfire_fm/checkpoints"><b>Download checkpoints</b></a> ·
+  <a href="https://huggingface.co/RAI-Lab/Wildfire-FM/tree/main/training"><b>Training/data loader</b></a> ·
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM#quick-start"><b>Load the model</b></a> ·
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM#task-contract-snapshot"><b>Inspect task-contract results</b></a>
 </p>
@@ -48,6 +49,7 @@ pretty_name: WildFIRE-FM
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM#release-navigation">Release Navigation</a> ·
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM#visual-tour">Visual Tour</a> ·
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM#quick-start">Quick Start</a> ·
+  <a href="https://huggingface.co/RAI-Lab/Wildfire-FM/tree/main/training">Training Pipeline</a> ·
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM#task-contract-snapshot">Task Snapshot</a> ·
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM#data-sources">Data Sources</a> ·
   <a href="https://huggingface.co/RAI-Lab/Wildfire-FM#how-to-cite">How to Cite</a>
@@ -79,6 +81,10 @@ pretty_name: WildFIRE-FM
     </td>
   </tr>
   <tr>
+    <td width="33%" valign="top">
+      <h3><a href="https://huggingface.co/RAI-Lab/Wildfire-FM/tree/main/training">Training/Data Loader</a></h3>
+      <p>Read the original cache builder, tensor assembly, tile sampler, full-map loader, and training loop.</p>
+    </td>
     <td width="33%" valign="top">
       <h3><a href="https://huggingface.co/RAI-Lab/Wildfire-FM#quick-start">Quick Start</a></h3>
       <p>Clone the repository, instantiate the model, load a seeded checkpoint, and run the artifact check.</p>
@@ -87,12 +93,12 @@ pretty_name: WildFIRE-FM
       <h3><a href="https://huggingface.co/RAI-Lab/Wildfire-FM#task-contract-snapshot">Task Snapshot</a></h3>
       <p>View the final-paper task-contract summary for occupancy, spread, retrieval, burned area, smoke, and heat.</p>
     </td>
+  </tr>
+  <tr>
     <td width="33%" valign="top">
       <h3><a href="https://huggingface.co/RAI-Lab/Wildfire-FM/tree/main/artifacts/results">Numeric Artifacts</a></h3>
       <p>Use sanitized CSV and JSON summaries that back the public model-card tables and figure previews.</p>
     </td>
-  </tr>
-  <tr>
     <td width="33%" valign="top">
       <h3><a href="https://huggingface.co/RAI-Lab/Wildfire-FM#visual-tour">Visual Tour</a></h3>
       <p>Browse final-paper previews for matching-rule sensitivity, selection regret, and task-form ranking changes.</p>
@@ -101,10 +107,6 @@ pretty_name: WildFIRE-FM
       <h3><a href="https://huggingface.co/RAI-Lab/Wildfire-FM/blob/main/data_sources/DATA_SOURCES.md">Data Sources</a></h3>
       <p>See the public provider links and roles for HRRR, FIRMS, LANDFIRE, WRC, LandScan, WFIGS, and MTBS.</p>
     </td>
-    <td width="33%" valign="top">
-      <h3><a href="https://huggingface.co/RAI-Lab/Wildfire-FM#repository-layout">Repository Layout</a></h3>
-      <p>Find where checkpoints, manifests, scripts, paper outputs, and release documentation live in this Hub repo.</p>
-    </td>
   </tr>
 </table>
 
@@ -193,6 +195,7 @@ This Hub repository is model-first. It provides the released weights, model-load
 | Training split | June-August 2024 train, September 2024 validation, October 2024 test |
 | Released checkpoints | 5 seeded PyTorch checkpoints |
 | Data access path | Source links and roles documented in `data_sources/` |
+| Training/data loader path | Original cache builder and PyTorch loader in `training/` |
 | Paper artifact path | Compiled PDF in `paper/`; manuscript source managed in the authors' paper workspace |
 
 ### Input Channel Contract
@@ -224,6 +227,7 @@ This Hub repository is a model release with paper-aligned reproducibility artifa
 | `models/wildfire_fm/` | Model code, manifests, checkpoint metadata | `models/wildfire_fm/modeling_unet.py` |
 | `models/wildfire_fm/input_channels.json` | Machine-readable 16-channel input contract | `models/wildfire_fm/input_channels.json` |
 | `models/wildfire_fm/checkpoints/` | Five seeded PyTorch checkpoints | `seed_1/best_firms_prauc.pt` |
+| `training/` | Original cache builder, data loader, tile sampler, and training loop | `train_cold_tiled_mainline.py` |
 | `paper/` | Compiled paper PDF | `wildfire_fm_evaluation_contracts.pdf` |
 | `assets/` | Hub-page visuals and final-paper previews | `selection_regret_final.png` |
 | `paper_outputs/figures/` | Selected final-paper figure PDFs | `fig_rank_heatmap1.pdf` |
@@ -319,6 +323,10 @@ WildFIRE-FM/
     modeling_unet.py
     checkpoint_manifest.json
     checkpoints/seed_*/best_firms_prauc.pt
+  training/
+    build_phase1_cache_regional_hrrr.py
+    train_cold_tiled_mainline.py
+    configs/
   paper/
     wildfire_fm_evaluation_contracts.pdf
   assets/