Skip to content

Commit 0b122c1

Browse files
committed
can support pytorch1.1.0
1 parent 2781d9a commit 0b122c1

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@ in **lib/network.py**.
99

1010

1111
## Environment
12-
- python 3.6
13-
- pytorch 0.4.1
12+
- python 3.6 or python 3.7.3
13+
- pytorch 0.4.1 or pytorch 1.1.0
1414

1515

1616
## Update Records
@@ -22,9 +22,9 @@ reason with me, and you can find it in [here](https://github.com/pytorch/pytorch
2222
4. For convenience, I split the **lib/non_local.py** into four python files, and move the
2323
old versions (**lib/non_loca.py** and **lib/non_local_simple_version.py**) into
2424
**lib/backup/**.
25-
5. modify the code to support pytorch 0.4.1, and move the code supporting pytorch 0.3.1 \
25+
5. Modify the code to support pytorch 0.4.1, and move the code supporting pytorch 0.3.1 \
2626
to **Non-Local_pytorch_0.3.1/**.
27-
27+
6. Test the code with pytorch 1.1.0 and it works.
2828

2929
## Running Steps
3030
1. Select the type of non-local block in **lib/network.py**.

0 commit comments

Comments
 (0)