Bladeren bron

[TSPP/PyTorch] Adjust formatting and version requirements

Izzy Putterman 3 jaren geleden
bovenliggende
commit
38f17716eb

+ 4 - 0
README.md

@@ -88,6 +88,10 @@ These examples, along with our NVIDIA deep learning software stack, are provided
 | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- |------------- |------------- |------------- |------------- |------------- |
 | [SE(3)-Transformer](https://github.com/NVIDIA/DeepLearningExamples/tree/master/DGLPyTorch/DrugDiscovery/SE3Transformer) | PyTorch  | Yes  | Yes  | Yes  | - | - | - | - | - | - |
 
+## Time-Series Forecasting
+| Models  | Framework | A100 | AMP | Multi-GPU | Multi-Node  | TRT  | ONNX  | Triton | DLC | NB | 
+| ------------- | ------------- | ------------- | ------------- | ------------- | ------------- |------------- |------------- |------------- |------------- |------------- |
+| [Temporal Fusion Transformer](https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Forecasting/TFT) | PyTorch  | Yes  | Yes  | Yes  | - | Yes | Yes | [Yes](https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Forecasting/TFT/triton) | Yes | - |
 
 ## NVIDIA support
 In each of the network READMEs, we indicate the level of support that will be provided. The range is from ongoing updates and improvements to a point-in-time release for thought leadership.

+ 2 - 1
Tools/PyTorch/TimeSeriesPredictionPlatform/README.md

@@ -418,7 +418,8 @@ We’re constantly refining and improving our performance on AI and HPC workload
 
 ### Changelog
 November 2021
-- Initial release
+- Initial release 
+
 July 2022
 - Reworked config structure
 - Added parallel execution

+ 0 - 1
Tools/PyTorch/TimeSeriesPredictionPlatform/requirements.txt

@@ -13,7 +13,6 @@
 # limitations under the License.
 
 pmdarima==1.8.0
-matplotlib==3.3.2
 wget==3.2
 hydra-core==1.1.1
 pyunpack==0.2.2

+ 1 - 0
Tools/PyTorch/TimeSeriesPredictionPlatform/triton/requirements.txt

@@ -21,3 +21,4 @@ pycuda>=2019.1.2
 PyYAML>=5.2
 tabulate>=0.8.7
 tqdm>=4.44.1
+polygraphy==0.36.2