NVIDIA · ChayanBhansali · Aug 7, 2024 · Aug 12, 2024
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/README.md b/Tools/PyTorch/TimeSeriesPredictionPlatform/README.md
@@ -153,7 +153,7 @@ For more information about how to get started with NGC containers, refer to the
 2. Enter the Deep Learning Examples TSPP repository:
 
 ```
-cd DeeplearningExamples/Tools/TimeSeriesPredictionPlatform
+cd DeeplearningExamples/Tools/PyTorch/TimeSeriesPredictionPlatform
 ```
 3. Copy the relevant temporal fusion transformer [code](https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Forecasting/TFT/modeling.py) to the TSPP:
 ```

diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/dcrnn.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/dcrnn.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/deepar.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/deepar.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/deepar_v2.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/deepar_v2.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/ensembling.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/ensembling.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/gnn.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/gnn.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/interpretability.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/interpretability.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/lstm.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/lstm.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/mtgnn.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/mtgnn.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/nbeats.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/nbeats.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/nhits.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/nhits.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/stat_models.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/stat_models.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/tft.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/tft.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/trivial_model.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/trivial_model.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/models/tspp_xgboost.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/models/tspp_xgboost.py
diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/triton/requirements.txt b/Tools/PyTorch/TimeSeriesPredictionPlatform/triton/requirements.txt
@@ -16,7 +16,7 @@ natsort>=7.0.0
 networkx==2.5
 numpy
 onnx>=1.8.0,<1.9.0
-onnxruntime-gpu==1.8.1
+onnxruntime-gpu==1.18.1
 pycuda>=2019.1.2
 PyYAML>=5.2
 tabulate>=0.8.7