Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
tianzhi0549 authored May 1, 2020
1 parent 85fc8bc commit bdbca6c
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -45,9 +45,8 @@ Name | inf. time | box AP | mask AP | download

For more models and information, please refer to MEInst [README.md](configs/MEInst-InstanceSegmentation/README.md).


Note that:
- Inference time for all projects is measured on a NVIDIA 1080Ti with batch size 1.

- APs are evluated on COCO2017 val split unless specified.

## Installation
Expand Down Expand Up @@ -98,7 +97,7 @@ OMP_NUM_THREADS=1 python tools/train_net.py \
OUTPUT_DIR training_dir/fcos_R_50_1x \
MODEL.WEIGHTS training_dir/fcos_R_50_1x/model_final.pth
```

Note that:
- The configs are made for 8-GPU training. To train on another number of GPUs, change the `--num-gpus`.
- If you want to measure the inference time, please change `--num-gpus` to 1.
- We set `OMP_NUM_THREADS=1` by default, which achieves the best speed on our machines, please change it as needed.
Expand Down

0 comments on commit bdbca6c

Please sign in to comment.