Add serving-oriented tiled inference and jittered training support
Browse files- training/README.md +12 -0
training/README.md
CHANGED
|
@@ -83,6 +83,13 @@ The training script reads `splits/train.csv`, `splits/val.csv`, and
|
|
| 83 |
`splits/test.csv` from `index_root`. Training uses 32-by-32 tiles sampled from
|
| 84 |
the cached time maps. Validation and test use full maps.
|
| 85 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 86 |
```bash
|
| 87 |
python training/train_cold_tiled_mainline.py \
|
| 88 |
--config training/configs/train_firewx_fm_seed7_template.json \
|
|
@@ -100,3 +107,8 @@ For inference, the important part is to reproduce the same 16-channel tensor
|
|
| 100 |
order. If a downstream environment already constructs `[channel, y, x]` tensors
|
| 101 |
in that order, it can use `models/wildfire_fm/modeling_unet.py` directly and
|
| 102 |
load one of the released seeded checkpoints.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 83 |
`splits/test.csv` from `index_root`. Training uses 32-by-32 tiles sampled from
|
| 84 |
the cached time maps. Validation and test use full maps.
|
| 85 |
|
| 86 |
+
For serving-oriented retraining, set `positive_tile_placement` to
|
| 87 |
+
`random_containing`. This samples positive tiles so the fire cell can appear at
|
| 88 |
+
different positions within the tile instead of always being centered. The
|
| 89 |
+
training script also supports `input_normalization`; use it to compute
|
| 90 |
+
train-split z-score statistics for continuous weather and static channels while
|
| 91 |
+
leaving masks and categorical fuel codes unscaled.
|
| 92 |
+
|
| 93 |
```bash
|
| 94 |
python training/train_cold_tiled_mainline.py \
|
| 95 |
--config training/configs/train_firewx_fm_seed7_template.json \
|
|
|
|
| 107 |
order. If a downstream environment already constructs `[channel, y, x]` tensors
|
| 108 |
in that order, it can use `models/wildfire_fm/modeling_unet.py` directly and
|
| 109 |
load one of the released seeded checkpoints.
|
| 110 |
+
|
| 111 |
+
Avoid non-overlapping 32-by-32 serving tiles. If full-map inference is not
|
| 112 |
+
available, use overlap-tiled inference with halo cropping or overlap blending.
|
| 113 |
+
The helper `models/wildfire_fm/tiled_inference.py` provides
|
| 114 |
+
`predict_probability_tiled` for this stitching pattern.
|