tpys commited on
Commit ·
dc45ef3
1
Parent(s): 1cff2ab
Describe diagnostic feedback without misrepresenting channel flow
Browse filesClarify that radiation-flux and precipitation channels stay in the fixed recurrent tensor but are zeroed before the next step. Document the zeroing stage in the rollout sequence.
Constraint: Preserve the model's fixed 85-channel recurrent interface
Rejected: Describe diagnostic fields as absent from feedback | the channels remain present even though their values are zeroed
Confidence: high
Scope-risk: narrow
Directive: Distinguish channel presence from value propagation when documenting diagnostic fields
Tested: git diff --cached --check; reviewed the data-details and rollout descriptions together
Not-tested: Inference execution; documentation-only correction
README.md
CHANGED
|
@@ -145,8 +145,9 @@ FuXi-2.1 operates on 85 channels per time step: 65 pressure-level channels
|
|
| 145 |
(5 variables × 13 levels) and 20 surface channels, plus static forcings supplied as
|
| 146 |
constant inputs. Most channels are **prognostic**: they are both input and output
|
| 147 |
channels and are fed back during rollout. Radiation fluxes and total precipitation are
|
| 148 |
-
**diagnostic** outputs from a dedicated decoder head
|
| 149 |
-
|
|
|
|
| 150 |
|
| 151 |
**Pressure-level variables** (5 vars × 13 levels = 65 channels):
|
| 152 |
|
|
@@ -202,8 +203,10 @@ fuxi-2.1/
|
|
| 202 |
|
| 203 |
1. **Load** pre-normalized input (2 frames at t-6h and t)
|
| 204 |
2. **Rollout** autoregressively: model outputs next 2-frame state, last frame is the prediction
|
| 205 |
-
3. **
|
| 206 |
-
|
|
|
|
|
|
|
| 207 |
|
| 208 |
The recurrence state stays on GPU throughout the rollout (no CPU round-trip per step).
|
| 209 |
|
|
|
|
| 145 |
(5 variables × 13 levels) and 20 surface channels, plus static forcings supplied as
|
| 146 |
constant inputs. Most channels are **prognostic**: they are both input and output
|
| 147 |
channels and are fed back during rollout. Radiation fluxes and total precipitation are
|
| 148 |
+
**diagnostic** outputs from a dedicated decoder head. Their channels remain present in
|
| 149 |
+
the recurrent input to preserve the fixed 85-channel layout, but their values are zeroed
|
| 150 |
+
before feedback so accumulated diagnostic fields are not propagated to the next step.
|
| 151 |
|
| 152 |
**Pressure-level variables** (5 vars × 13 levels = 65 channels):
|
| 153 |
|
|
|
|
| 203 |
|
| 204 |
1. **Load** pre-normalized input (2 frames at t-6h and t)
|
| 205 |
2. **Rollout** autoregressively: model outputs next 2-frame state, last frame is the prediction
|
| 206 |
+
3. **Prepare recurrence**: retain the prediction for output, then zero radiation-flux and
|
| 207 |
+
precipitation channels in the recurrent state before the next step
|
| 208 |
+
4. **Denormalize** each prediction: `output = output * std + mean`, then `expm1` for precipitation
|
| 209 |
+
5. **Save** each step as NetCDF with geographic coordinates
|
| 210 |
|
| 211 |
The recurrence state stays on GPU throughout the rollout (no CPU round-trip per step).
|
| 212 |
|