Commit 69e17d85 authored by Yuxin Wu's avatar Yuxin Wu

update docs

parent d5306602
# Performance Tuning # Performance Tuning
Here's a list of things you can do when your training is slow: Here's a list of things you can do when your training is slow.
And if you're going to open an issue about slow training, PLEASE do them and include your findings.
## Figure out the bottleneck ## Figure out the bottleneck
......
...@@ -44,10 +44,13 @@ To evaluate the performance (pretrained models can be downloaded in [model zoo]( ...@@ -44,10 +44,13 @@ To evaluate the performance (pretrained models can be downloaded in [model zoo](
## Results ## Results
Mean Average Precision @IoU=0.50:0.95: Trained on trainval35k and evaluated on minival, got the following results:
mAP@IoU=0.50:0.95:
+ trainval35k/minival, FASTRCNN_BATCH=256: 34.2. Takes 49h on 8 TitanX. |Backbone | `FASTRCNN_BATCH` | mAP | Time |
+ trainval35k/minival, FASTRCNN_BATCH=64: 33.0. Takes 22h on 8 P100. | - | - | - | - |
| Res50 | 256 | 34.4 | 49h on 8 TitanX |
| Res50 | 64 | 33.0 | 22h on 8 P100 |
The hyperparameters are not carefully tuned. You can probably get better performance by e.g. training longer. The hyperparameters are not carefully tuned. You can probably get better performance by e.g. training longer.
......
...@@ -17,6 +17,7 @@ RESNET_NUM_BLOCK = [3, 4, 6, 3] # resnet50 ...@@ -17,6 +17,7 @@ RESNET_NUM_BLOCK = [3, 4, 6, 3] # resnet50
# preprocessing -------------------- # preprocessing --------------------
SHORT_EDGE_SIZE = 600 SHORT_EDGE_SIZE = 600
MAX_SIZE = 1024 MAX_SIZE = 1024
# alternative (better) setting: 800, 1333
# anchors ------------------------- # anchors -------------------------
ANCHOR_STRIDE = 16 ANCHOR_STRIDE = 16
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment