Skip to content

Commit

Permalink
add how to switch pytorch 1.0 to 0.4
Browse files Browse the repository at this point in the history
  • Loading branch information
CharlesShang committed Dec 11, 2018
1 parent 1a1c100 commit b4d2fe0
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,11 @@

dout = dpooling(input, rois)
```
### Note
Now the master branch is for pytorch 1.0 (new ATen API), you can switch back to pytorch 0.4 with,
```bash
git checkout pytorch_0.4
```

### Known Issues:

Expand Down

0 comments on commit b4d2fe0

Please sign in to comment.