Skip to content

Commit cf1bb34

Browse files
committed
update explanation for speed and add cython in requirements
1 parent 6614b7a commit cf1bb34

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

README.MD

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,9 @@ VGG16 train on `trainval` and test on `test` split.
4343
| This[1] | TITAN Xp | 14-15 fps | 6 fps |
4444
| [pytorch-faster-rcnn](https://github.com/ruotianluo/pytorch-faster-rcnn) | TITAN Xp | 15-17fps | 6fps |
4545

46-
[1]: make sure you install cupy correctly and only one program run on the GPU. The training speed is sensitive to your gpu status. Morever it's slow in the start of the program.
47-
It could be even faster by removing visualization, logging, averaging loss etc.
46+
[1]: make sure you install cupy correctly and only one program run on the GPU. The training speed is sensitive to your gpu status. see [troubleshooting](troubleshooting) for more info. Morever it's slow in the start of the program.
47+
48+
It could be faster by removing visualization, logging, averaging loss etc.
4849
## 3. Install dependencies
4950

5051
requires python3 and PyTorch 0.3
@@ -184,7 +185,11 @@ If you're in China and encounter problem with visdom (i.e. timeout, blank screen
184185

185186
You don't need to build from source, modifying related files would be OK.
186187

188+
- About the speed
189+
190+
One strange thing is that, even the code doesn't use chainer, but if I remove `from chainer import cuda`, the speed drops a lot (train 6.5->6.1,test 14.5->10), because Chainer replaces the default allocator of CuPy by its memory pool implementation. But ever since V4.0, cupy use memory pool as default. However you need to build from souce if you are gona use the latest version of cupy (uninstall cupy -> git clone -> git checkout v4.0 -> setup.py install) @_@
187191

192+
Another simple fix: add `from chainer import cuda` at the begining of `train.py`. in such case,you'll need to `pip install chainer` first.
188193
## More
189194

190195
- [ ] training on coco

requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,4 +6,5 @@ fire
66
pprint
77
matplotlib
88
ipdb
9+
cython
910
git+https://github.com/pytorch/tnt.git@master

0 commit comments

Comments
 (0)