wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-21 12:50:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

scale input to network if flag is set

Browse Source
This commit is contained in:
Kashif Rasul
2020-02-01 18:04:01 +01:00
parent bab7adbc7e
commit 8fa0297ab7
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/tempflow/tempflow_network.py
+4 -1
View File
@@ -75,7 +75,10 @@ class TempFlowTrainingNetwork(nn.Module):
num_embeddings=self.target_dim, embedding_dim=self.embed_dim
)
self.scaler = MeanScaler(keepdim=True)
if self.scaling:
self.scaler = MeanScaler(keepdim=True)
else:
self.scaler = NOPScaler(keepdim=True)
@staticmethod
def get_lagged_subsequences(
pts/model/transformer_tempflow/transformer_tempflow_network.py
+4 -1
View File
@@ -82,7 +82,10 @@ class TransformerTempFlowTrainingNetwork(nn.Module):
num_embeddings=self.target_dim, embedding_dim=self.embed_dim
)
self.scaler = MeanScaler(keepdim=True)
if self.scaling:
self.scaler = MeanScaler(keepdim=True)
else:
self.scaler = NOPScaler(keepdim=True)
# mask
self.register_buffer(