maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Update README.md

Browse source
This commit is contained in:
nv-kkudrynski 2021-01-08 23:12:26 +01:00 committed by GitHub
parent 3d0d45b409
commit 4eb551ded7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
PyTorch/Recommendation/DLRM/triton/README.md
View file

@ -25,7 +25,7 @@ The very first step of deployment is to acquire trained checkpoint and model con
checkpoint. Default model configuration are stored inside `dlrm/config` directory.
**Currently, our implementation only supports TorchScript deployment for models that fit into the memory of a single GPU.**
You can read more about training DLRM models on different dataset configurations based on frequency threshold in the preprocessing step in [README](https://gitlab-master.nvidia.com/dl/JoC/dlrm_pyt#preprocess-with-spark).
You can read more about training DLRM models on different dataset configurations based on frequency threshold in the preprocessing step in [README](https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Recommendation/DLRM/README.md#preprocess-with-spark).
#### Inference container