Merge pull request #601 from NVIDIA/mod_doc_fix

narendasan · web-flow · commit 19354bcd29f8 · 2021-08-24T09:53:44.000-06:00
diff --git a/tests/modules/README.md b/tests/modules/README.md
@@ -1,6 +1,6 @@
 # Module Level Tests
 
-These tests focus on integration testing of the backend, using the public apis. They require TorchScript modules saved to files. You can download the set of modules used by default in the tests by running hub.py (you need PyTorch 1.4.0) installed. 
+These tests focus on integration testing of the backend, using the public apis. They require TorchScript modules saved to files. You can download the set of modules used by default in the tests by running hub.py (you need PyTorch 1.9.0 installed with cuda-11.1).
 
 ## Running Tests
 
diff --git a/tests/modules/requirements.txt b/tests/modules/requirements.txt
@@ -1 +1,3 @@
+-f https://download.pytorch.org/whl/torch_stable.html
 timm==v0.4.12
+torch==1.9.0+cu111

Original file line number	Diff line number	Diff line change
`@@ -1 +1,3 @@`
	`1`	`+-f https://download.pytorch.org/whl/torch_stable.html`
`1`	`2`	`timm==v0.4.12`
	`3`	`+torch==1.9.0+cu111`