saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-27 20:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove previous temporary fixes and address TODOs (#9020)

Browse source
This commit is contained in:
baijumeswani 2021-09-13 10:10:07 -07:00 committed by GitHub
parent 011cb8fd48
commit 1422a9ba6b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 0 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
onnxruntime/test/python/onnxruntime_test_ort_trainer.py
View file

@ -249,23 +249,6 @@ class MNISTWrapper():
return ModelDescription([input_desc, label_desc], [loss_desc, probability_desc])
def get_loaders(self):
# TODO: Remove this temporary fix for urllib.error.HTTPError: HTTP Error 403: Forbidden
# once a more permanent solution can be found.
# Fix as per https://github.com/pytorch/vision/issues/1938#issuecomment-789986996
from six.moves import urllib
opener = urllib.request.build_opener()
opener.addheaders = [('User-agent', 'Mozilla/5.0')]
urllib.request.install_opener(opener)
# TODO: Remove this temporary fix when the issue https://github.com/pytorch/vision/issues/3549 is resolved
# Resource http://yann.lecun.com/exdb/mnist/ is not available
datasets.MNIST.resources = [
('https://ossci-datasets.s3.amazonaws.com/mnist/train-images-idx3-ubyte.gz', 'f68b3c2dcbeaaa9fbdd348bbdeb94873'),
('https://ossci-datasets.s3.amazonaws.com/mnist/train-labels-idx1-ubyte.gz', 'd53e105ee54ea40749a09fcbcd1e9432'),
('https://ossci-datasets.s3.amazonaws.com/mnist/t10k-images-idx3-ubyte.gz', '9fb629c4189551a2d022fa330f9573f3'),
('https://ossci-datasets.s3.amazonaws.com/mnist/t10k-labels-idx1-ubyte.gz', 'ec29112dd5afa0611ce80d1b7f02629c')
]
args_batch_size = 64
args_test_batch_size = 1000

2
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements-torch1.9.0_rocm.txt
View file

@ -1,5 +1,3 @@
# TODO: Remove pinned version of Pillow of 8.2.0 when 8.3.0 release regression issue https://github.com/python-pillow/Pillow/issues/5571 is resolved
Pillow==8.2.0
# transformers requires sklearn
--pre
-f https://download.pytorch.org/whl/rocm4.2/torch_stable.html

2
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch1.9.0_cu10.2.txt
View file

@ -1,5 +1,3 @@
# TODO: Remove pinned version of Pillow of 8.2.0 when 8.3.0 release regression issue https://github.com/python-pillow/Pillow/issues/5571 is resolved
Pillow==8.2.0
--pre
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.9.0+cu102

2
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch1.9.0_cu11.1.txt
View file

@ -1,5 +1,3 @@
# TODO: Remove pinned version of Pillow of 8.2.0 when 8.3.0 release regression issue https://github.com/python-pillow/Pillow/issues/5571 is resolved
Pillow==8.2.0
--pre
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.9.0+cu111

2
tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch1.9.0_cu11.4.txt
View file

@ -1,5 +1,3 @@
# TODO: Remove pinned version of Pillow of 8.2.0 when 8.3.0 release regression issue https://github.com/python-pillow/Pillow/issues/5571 is resolved
Pillow==8.2.0
# TODO: set to cu114 once it is available from pytorch, and to match torchvision and torchtext versions
--pre
-f https://download.pytorch.org/whl/torch_stable.html