Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
endernewton committed May 22, 2017
1 parent d339928 commit 271d6c5
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -55,9 +55,10 @@ Additional features not mentioned in the [report](https://arxiv.org/pdf/1702.021
2. Update your -arch in setup script to match your GPU
```Shell
cd tf-faster-rcnn/lib
# Check the GPU architecture
vim setup.py
# Check the GPU architecture, if you are using Pascal/Maxwell arch, sm_52 should work
```
**Note**: I only have access to TitanX (both Maxwell and Pascal) and sm_52 seems to work fine on those machines. You are welcome to contribute the settings on your end if you have made the code work properly on other GPUs.

3. Build the Cython modules
```Shell
Expand Down

0 comments on commit 271d6c5

Please sign in to comment.