@@ -47,7 +47,6 @@ def __init__(
4747 add_image_embedding_channel : bool = False ,
4848 include_gsp_yield_history : bool = True ,
4949 include_sun : bool = True ,
50- include_gsp : bool = True ,
5150 embedding_dim : Optional [int ] = 16 ,
5251 forecast_minutes : int = 30 ,
5352 history_minutes : int = 60 ,
@@ -108,7 +107,6 @@ def __init__(
108107 interval_minutes: The interval between each sample of the target data
109108 wind_encoder: Encoder for wind data
110109 wind_history_minutes: Length of recent wind data used as input.
111- include_gsp: Whether to include GSP data in the model
112110 pv_interval_minutes: The interval between each sample of the PV data
113111 sat_interval_minutes: The interval between each sample of the satellite data
114112 sensor_interval_minutes: The interval between each sample of the sensor data
@@ -119,7 +117,6 @@ def __init__(
119117 self .include_nwp = nwp_encoders_dict is not None and len (nwp_encoders_dict ) != 0
120118 self .include_pv = pv_encoder is not None
121119 self .include_sun = include_sun
122- self .include_gsp = include_gsp
123120 self .include_wind = wind_encoder is not None
124121 self .include_sensor = sensor_encoder is not None
125122 self .embedding_dim = embedding_dim
@@ -210,7 +207,7 @@ def __init__(
210207 if wind_history_minutes is None :
211208 wind_history_minutes = history_minutes
212209
213- self .wind_encoder = wind_encoder (sequence_length = self . history_len_30 )
210+ self .wind_encoder = wind_encoder (sequence_length = wind_history_minutes // interval_minutes )
214211
215212 # Update num features
216213 fusion_input_features += self .wind_encoder .out_features
@@ -219,7 +216,7 @@ def __init__(
219216 if sensor_history_minutes is None :
220217 sensor_history_minutes = history_minutes
221218
222- self .sensor_encoder = sensor_encoder (sequence_length = self . history_len_30 )
219+ self .sensor_encoder = sensor_encoder (sequence_length = sensor_history_minutes // sensor_interval_minutes )
223220
224221 # Update num features
225222 fusion_input_features += self .sensor_encoder .out_features
0 commit comments