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.
- {patchfm-3.1.0/src/patchfm.egg-info → patchfm-3.1.1}/PKG-INFO +1 -1
- {patchfm-3.1.0 → patchfm-3.1.1}/pyproject.toml +1 -1
- {patchfm-3.1.0 → patchfm-3.1.1}/src/patchfm/inference/forecaster.py +2 -2
- {patchfm-3.1.0 → patchfm-3.1.1}/src/patchfm/inference/modules.py +3 -3
- {patchfm-3.1.0 → patchfm-3.1.1/src/patchfm.egg-info}/PKG-INFO +1 -1
- {patchfm-3.1.0 → patchfm-3.1.1}/LICENSE +0 -0
- {patchfm-3.1.0 → patchfm-3.1.1}/README.md +0 -0
- {patchfm-3.1.0 → patchfm-3.1.1}/setup.cfg +0 -0
- {patchfm-3.1.0 → patchfm-3.1.1}/src/patchfm/__init__.py +0 -0
- {patchfm-3.1.0 → patchfm-3.1.1}/src/patchfm/configs/model_config.py +0 -0
- {patchfm-3.1.0 → patchfm-3.1.1}/src/patchfm.egg-info/SOURCES.txt +0 -0
- {patchfm-3.1.0 → patchfm-3.1.1}/src/patchfm.egg-info/dependency_links.txt +0 -0
- {patchfm-3.1.0 → patchfm-3.1.1}/src/patchfm.egg-info/requires.txt +0 -0
- {patchfm-3.1.0 → patchfm-3.1.1}/src/patchfm.egg-info/top_level.txt +0 -0
|
@@ -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=
|
|
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=
|
|
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.
|
|
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=
|
|
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=
|
|
326
|
+
hid_dim=4 * self.d_model,
|
|
327
327
|
out_dim=self.patch_len * self.n_quantiles
|
|
328
328
|
)
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|