Revert "Merge pull request #747 from AIStream-Peelout/crossformer_fixes"

This reverts commit 224b67c, reversing changes made to 9074dc0.
AIStream-Peelout · May 7, 2024 · 2dd6d0d · 2dd6d0d
1 parent 224b67c
commit 2dd6d0d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 7 deletions.
diff --git a/flood_forecast/transformer_xl/cross_former.py b/flood_forecast/transformer_xl/cross_former.py
@@ -19,7 +19,6 @@ def __init__(
         e_layers=3,
         dropout=0.0,
         baseline=False,
-        n_targs=None,
         device=torch.device("cuda:0"),
     ):
         """Crossformer: Transformer Utilizing Cross-Dimension Dependency for Multivariate Time Series Forecasting.
@@ -58,7 +57,6 @@ def __init__(
         self.out_len = forecast_length
         self.seg_len = seg_len
         self.merge_win = win_size
-        self.n_targs = n_time_series if n_targs is None else n_targs
 
         self.baseline = baseline
 
@@ -128,9 +126,7 @@ def forward(self, x_seq: torch.Tensor):
         )
         predict_y = self.decoder(dec_in, enc_out)
 
-        result = base + predict_y[:, : self.out_len, :]
-        res = result[:, :, :self.n_targs]
-        return res
+        return base + predict_y[:, : self.out_len, :]
 
 
 class SegMerging(nn.Module):

diff --git a/flood_forecast/transformer_xl/multi_head_base.py b/flood_forecast/transformer_xl/multi_head_base.py
@@ -5,7 +5,7 @@
 
 
 class MultiAttnHeadSimple(torch.nn.Module):
-    """A simple multi-head attention model inspired by Vas.,wani et al."""
+    """A simple multi-head attention model inspired by Vaswani et al."""
 
     def __init__(
             self,

diff --git a/setup.py b/setup.py
@@ -16,7 +16,7 @@
 
 setup(
     name='flood_forecast',
-    version='1.0001dev',
+    version='1.001dev',
     packages=[
         'flood_forecast',
         'flood_forecast.transformer_xl',