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

typos

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-01-29 14:25:22 +01:00
parent c7bea1e084
commit 7a7ddc5997
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/transformer_tempflow/transformer_tempflow_estimator.py
+4 -4
View File
@@ -163,8 +163,8 @@ class TransformerTempFlowEstimator(PTSEstimator):
]
)
def create_training_network(self, device: torch.device) -> TempFlowTrainingNetwork:
return TempFlowTrainingNetwork(
def create_training_network(self, device: torch.device) -> TransformerTempFlowTrainingNetwork:
return TransformerTempFlowTrainingNetwork(
input_size=self.input_size,
target_dim=self.target_dim,
num_heads=self.num_heads,
@@ -192,10 +192,10 @@ class TransformerTempFlowEstimator(PTSEstimator):
def create_predictor(
self,
transformation: Transformation,
trained_network: TempFlowTrainingNetwork,
trained_network: TransformerTempFlowTrainingNetwork,
device: torch.device,
) -> PTSPredictor:
prediction_network = TempFlowPredictionNetwork(
prediction_network = TransformerTempFlowPredictionNetwork(
input_size=self.input_size,
target_dim=self.target_dim,
num_heads=self.num_heads,