yx21e commited on
Commit
13bf22d
·
verified ·
1 Parent(s): 9ac0772

Add serving-oriented tiled inference and jittered training support

Browse files
Files changed (1) hide show
  1. 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.