diff --git a/README.md b/README.md index c54cec4b364286e821fdc8fbe2bcc4e4b0f9154e..499b63214c27fc14000d8a9d1731c1ec29cf2160 100644 --- a/README.md +++ b/README.md @@ -52,3 +52,6 @@ nvidia-docker run --gpus=all --rm --shm-size 5g --network=host --name dhruva-him We provide a sample ipython notebook that shows how to concurrently request the client for translations. Prerequisites: `pip install "tritonclient[all]" tqdm numpy wonderwords` + +## Data and Model Training +Please refer to the training-code folder and the README within for the links to the data and checkout the Ipython notebook for an annotated version of the training script.