wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-06-27 18:06:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed typo

Browse Source
This commit is contained in:
Kashif Rasul
2021-02-18 12:11:48 +01:00
committed by GitHub
parent 41b3d7f883
commit 9238fff0f5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/transformer/transformer_estimator.py
+1 -1
View File
@@ -251,7 +251,7 @@ class TransformerEstimator(PyTorchEstimator):
copy_parameters(trained_network, prediction_network)
input_names = get_module_forward_input_names(prediction_network)
prediction_splitter = self._create_instance_splitter("test")
prediction_splitter = self.create_instance_splitter("test")
return PyTorchPredictor(
input_transform=transformation + prediction_splitter,