diff --git a/on_device_training/desktop/csharp/masked_language_modeling/requirements.txt b/on_device_training/desktop/csharp/masked_language_modeling/requirements.txt new file mode 100644 index 00000000..38cbf9c1 --- /dev/null +++ b/on_device_training/desktop/csharp/masked_language_modeling/requirements.txt @@ -0,0 +1,4 @@ +numpy==1.26.3 +torch +transformers +datasets diff --git a/on_device_training/desktop/python/mnist.ipynb b/on_device_training/desktop/python/mnist.ipynb index 4338952c..1ad1f507 100644 --- a/on_device_training/desktop/python/mnist.ipynb +++ b/on_device_training/desktop/python/mnist.ipynb @@ -26,9 +26,7 @@ "source": [ "#### Importing libraries\n", "\n", - "Make sure to install onnxruntime-training's nightly version.\n", - "\n", - "```pip install onnxruntime-training```" + "Make sure to install onnxruntime-training's nightly version. Check [our website (click Optimize Training)](https://onnxruntime.ai/getting-started) for instructions on downloading the onnxruntime-training nightly Python package." ] }, { diff --git a/on_device_training/web/offline-step/README.md b/on_device_training/web/offline-step/README.md index 5ce56790..7ae2849d 100644 --- a/on_device_training/web/offline-step/README.md +++ b/on_device_training/web/offline-step/README.md @@ -5,6 +5,7 @@ This step is the same regardless of the method used to import the ONNXRuntime-we ## Set-up Install dependencies. This step requires onnxruntime-training-cpu>=1.17.0. +Check [our website (click Optimize Training)](https://onnxruntime.ai/getting-started) for instructions on downloading the onnxruntime-training nightly Python package. ``` pip install -r requirements.txt pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT/pypi/simple/ onnxruntime-training-cpu