From eaacf0456b6d560b81580f891b6d0adb07aa0272 Mon Sep 17 00:00:00 2001 From: carzh Date: Tue, 25 Jun 2024 14:34:10 -0700 Subject: [PATCH 1/2] updated installation instructions --- on_device_training/desktop/python/mnist.ipynb | 4 +--- on_device_training/web/offline-step/README.md | 1 + 2 files changed, 2 insertions(+), 3 deletions(-) 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 From cebaf5cb5077007163d16720f139c5c847df66e6 Mon Sep 17 00:00:00 2001 From: carzh Date: Mon, 1 Jul 2024 11:29:21 -0700 Subject: [PATCH 2/2] added requirements.txt for the offline step for mobilebert example --- .../desktop/csharp/masked_language_modeling/requirements.txt | 4 ++++ 1 file changed, 4 insertions(+) create mode 100644 on_device_training/desktop/csharp/masked_language_modeling/requirements.txt 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