Rollout update - #5

Open

grassesi wants to merge 14 commits into

SavvasMel:developfrom

grassesi:rollout_update

Commits on Aug 11, 2026

Let BatchSamples stand in for ModelOutput
grassesi
committed
dc0d238
Make ModelOutput describe which forecast steps it holds

grassesi
and
claude
committed
3c31bd7
Index predictions through ModelOutput.chunk_idx/batch_idx

grassesi
and
claude
committed
f6121c9
Run pushforward steps under torch.no_grad()

grassesi
and
claude
committed
5a4f80e
Abstract rollout intialization from either sources or past saved predictions.
grassesi
committed
5070497
Use new abstraction to intialize rollout in `Model.forward`
grassesi
committed
0c4da71
Iterate over forecast_steps in `Model.forward`
grassesi
committed
0091daf
use source_samples instead of batch in `Model.forward`
grassesi
committed
07d07e3
Change interface of `Model.forward` to allow incremental rollout.
grassesi
committed
2b9f561
Spoofed targets no longer depend on non-empty predictions

grassesi
and
claude
committed
5fb0d8c
Pass the forecast steps to OutputBatchData instead of deriving them

grassesi
and
claude
committed
3595841
Write validation output for the forecast steps the chunk describes

grassesi
and
claude
committed
2e26f26
Move the validation rollout and output writing into one method

grassesi
and
claude
committed
a3129b0
Run the validation rollout in chunks of forecast.chunk_size

grassesi
and
claude
committed
f600647