Update README.md

AlexHex7 · May 6, 2021 · 1fdae52 · 1fdae52
1 parent 272fcf3
commit 1fdae52
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -12,7 +12,7 @@
 
 ## Environment
 - python 3.7.7
-- pytorch 1.4.0 (>=1.2.0, 1.7.0 works too)
+- pytorch 1.4.0 (>=1.2.0, 1.8.1 works too)
 - opencv 4.2.0.34 (others work too)
 
 ## Visualization
@@ -80,6 +80,8 @@ to **Non-Local_pytorch_0.3.1/**.
 
 11. Verify that the code works well in **pytorch 1.7.0**.
 
+12. Verify that the code works well in **pytorch 1.8.1**.
+
 ## Todo
 - Experiments on Charades dataset.
 - Experiments on COCO dataset.