patchfm 3.1.0__tar.gz → 3.1.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: patchfm
3
- Version: 3.1.0
3
+ Version: 3.1.1
4
4
  Summary: a Foundation Model for Univariate Time Series Forecasting
5
5
  Author-email: Samy-Melwan Vilhes <samy-melwan.vilhes@insa-rouen.fr>
6
6
  License: MIT License
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "patchfm"
3
- version = "3.1.0"
3
+ version = "3.1.1"
4
4
  authors = [
5
5
  { name="Samy-Melwan Vilhes", email="samy-melwan.vilhes@insa-rouen.fr" },
6
6
  ]
@@ -46,7 +46,7 @@ class Forecaster(nn.Module):
46
46
  self.revin = CausalRevIN()
47
47
  self.proj_embedding = ResidualBlock(
48
48
  in_dim=self.patch_len,
49
- hid_dim=2 * self.patch_len,
49
+ hid_dim=4 * self.d_model,
50
50
  out_dim=self.d_model
51
51
  )
52
52
  self.transformer_encoder = TransformerEncoder(
@@ -56,7 +56,7 @@ class Forecaster(nn.Module):
56
56
  )
57
57
  self.proj_output = ResidualBlock(
58
58
  in_dim=self.d_model,
59
- hid_dim=2 * self.d_model,
59
+ hid_dim=4 * self.d_model,
60
60
  out_dim=self.patch_len * self.n_quantiles
61
61
  )
62
62
 
@@ -175,7 +175,7 @@ class ResidualBlock(nn.Module):
175
175
  self.hidden_layer = nn.Linear(in_dim, hid_dim)
176
176
  self.output_layer = nn.Linear(hid_dim, out_dim)
177
177
  self.residual_layer = nn.Linear(in_dim, out_dim)
178
- self.act = nn.ReLU()
178
+ self.act = nn.SiLU()
179
179
 
180
180
  def forward(self, x):
181
181
  hid = self.act(self.hidden_layer(x))
@@ -313,7 +313,7 @@ class PatchFM(nn.Module, PyTorchModelHubMixin):
313
313
  self.revin = CausalRevIN()
314
314
  self.proj_embedding = ResidualBlock(
315
315
  in_dim=self.patch_len,
316
- hid_dim=2 * self.patch_len,
316
+ hid_dim=4 * self.d_model,
317
317
  out_dim=self.d_model
318
318
  )
319
319
  self.transformer_encoder = TransformerEncoder(
@@ -323,6 +323,6 @@ class PatchFM(nn.Module, PyTorchModelHubMixin):
323
323
  )
324
324
  self.proj_output = ResidualBlock(
325
325
  in_dim=self.d_model,
326
- hid_dim=2 * self.d_model,
326
+ hid_dim=4 * self.d_model,
327
327
  out_dim=self.patch_len * self.n_quantiles
328
328
  )
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: patchfm
3
- Version: 3.1.0
3
+ Version: 3.1.1
4
4
  Summary: a Foundation Model for Univariate Time Series Forecasting
5
5
  Author-email: Samy-Melwan Vilhes <samy-melwan.vilhes@insa-rouen.fr>
6
6
  License: MIT License
File without changes
File without changes
File without changes
File without changes