Logic/TSlib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(xPatch): remove redundant encoder argument from season_net call

Browse Source
This commit is contained in:
gameloader
2025-09-10 16:03:43 +08:00
parent 1044c60fe7
commit ff987da4c6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
models/xPatch_SparseChannel.py
View File

@ -92,7 +92,7 @@ class Model(nn.Module):
seasonal_init, trend_init = self.decomp(x_enc) seasonal_init, trend_init = self.decomp(x_enc)
# Season stream # Season stream
y_season = self.season_net(seasonal_init, encoder=self.season_encoder) # [B, C, pred_len] y_season = self.season_net(seasonal_init) # [B, C, pred_len]
# Trend stream # Trend stream
B, L, C = trend_init.shape B, L, C = trend_init.shape
@ -127,7 +127,7 @@ class Model(nn.Module):
seasonal_init, trend_init = self.decomp(x_enc) seasonal_init, trend_init = self.decomp(x_enc)
# Season stream # Season stream
y_season = self.season_net(seasonal_init, encoder=self.season_encoder) # [B, C, pred_len] y_season = self.season_net(seasonal_init) # [B, C, pred_len]
# print("shape:", trend_init.shape) # print("shape:", trend_init.shape)
# Trend stream # Trend stream