diff --git a/MODEL_ZOO.md b/MODEL_ZOO.md index a72e660930fb85c8a82b88ffd04a76bc0cdf0079..377157c1deceda89748676f15b17d41bae813b9e 100644 --- a/MODEL_ZOO.md +++ b/MODEL_ZOO.md @@ -10,11 +10,17 @@ ### Software environment - Python 3.6 / 3.7 -- PyTorch 0.4.1 +- PyTorch 1.0 - CUDA 9.0.176 - CUDNN 7.0.4 - NCCL 2.1.15 +Note: The train time was measured with PyTorch 0.4.1. We will update it later, which should be about 0.02s ~ 0.05s faster. + +## Mirror sites + +We use AWS as the main site to host our model zoo, and maintain a mirror on aliyun. +You can replace `https://s3.ap-northeast-2.amazonaws.com` with `https://open-mmlab.oss-cn-beijing.aliyuncs.com` in model urls. ## Common settings diff --git a/README.md b/README.md index 21ff0ffd4b7a9cb1504d15e30a216ece3fa72427..919f9e4eb3fdd355100b23a05152b88e4d054e9b 100644 --- a/README.md +++ b/README.md @@ -3,6 +3,9 @@ ## Introduction +The master branch works with **PyTorch 1.0**. If you would like to use PyTorch 0.4.1, +please checkout to the [pytorch-0.4.1](https://github.com/open-mmlab/mmdetection/tree/pytorch-0.4.1) branch. + mmdetection is an open source object detection toolbox based on PyTorch. It is a part of the open-mmlab project developed by [Multimedia Laboratory, CUHK](http://mmlab.ie.cuhk.edu.hk/). @@ -36,6 +39,9 @@ This project is released under the [Apache 2.0 license](LICENSE). ## Updates +v0.6rc0(06/02/2019) +- Migrate to PyTorch 1.0. + v0.5.7 (06/02/2019) - Add support for Deformable ConvNet v2. (Many thanks to the authors and [@chengdazhi](https://github.com/chengdazhi)) - This is the last release based on PyTorch 0.4.1. diff --git a/setup.py b/setup.py index 879c1e32590f0f941050081df5c25e266b13dd0e..724adf38b7c821d62595f30892d48742ccda2e73 100644 --- a/setup.py +++ b/setup.py @@ -11,8 +11,8 @@ def readme(): MAJOR = 0 -MINOR = 5 -PATCH = 7 +MINOR = 6 +PATCH = 'rc0' SUFFIX = '' SHORT_VERSION = '{}.{}.{}{}'.format(MAJOR, MINOR, PATCH, SUFFIX)