Adding residual connection to latent space forecasting.

clessig · clessig · commit a093765355fd · 2025-04-03T08:30:07.000+02:00
diff --git a/src/weathergen/model/model.py b/src/weathergen/model/model.py
@@ -593,7 +593,7 @@ def forward(self, model_params, batch, forecast_steps):
                 )
             ]
 
-            tokens = self.forecast(model_params, tokens)
+            tokens = tokens + self.forecast(model_params, tokens)
 
         # prediction for final step
         preds_all += [

Original file line number	Diff line number	Diff line change
`@@ -593,7 +593,7 @@ def forward(self, model_params, batch, forecast_steps):`
`593`	`593`	`)`
`594`	`594`	`]`
`595`	`595`
`596`		`- tokens = self.forecast(model_params, tokens)`
	`596`	`+ tokens = tokens + self.forecast(model_params, tokens)`
`597`	`597`
`598`	`598`	`# prediction for final step`
`599`	`599`	`preds_all += [`