mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmsegmentation/configs/fp16
History
sennnnn d35fbbdb47
[Enhancement] Add Dev tools to boost develop (#798) 
* Modify default work dir when training.

* Refactor gather_models.py.

* Add train and test matching list.

* Regression benchmark list.

* lower readme name to upper readme name.

* Add url check tool and model inference test tool.

* Modify tool name.

* Support duplicate mode of log json url check.

* Add regression benchmark evaluation automatic tool.

* Add train script generator.

* Only Support script running.

* Add evaluation results gather.

* Add exec Authority.

* Automatically make checkpoint root folder.

* Modify gather results save path.

* Coarse-grained train results gather tool.

* Complete benchmark train script.

* Make some little modifications.

* Fix checkpoint urls.

* Fix unet checkpoint urls.

* Fix fast scnn & fcn checkpoint url.

* Fix fast scnn checkpoint urls.

* Fix fast scnn url.

* Add differential results calculation.

* Add differential results of regression benchmark train results.

* Add an extra argument to select model.

* Update nonlocal_net & hrnet checkpoint url.

* Fix checkpoint url of hrnet and Fix some tta evaluation results and modify gather models tool.

* Modify fast scnn checkpoint url.

* Resolve new comments.

* Fix url check status code bug.

* Resolve some comments.

* Modify train scripts generator.

* Modify work_dir of regression benchmark results.

* model gather tool modification.
 		2021-09-02 09:44:51 -07:00
..
README.md [Enhancement] Add Dev tools to boost develop (#798) 2021-09-02 09:44:51 -07:00
deeplabv3_r101-d8_512x1024_80k_fp16_cityscapes.py [Bug fix] fixed fp16 inference (#497) 2021-04-24 09:57:33 -07:00
deeplabv3plus_r101-d8_512x1024_80k_fp16_cityscapes.py [Bug fix] fixed fp16 inference (#497) 2021-04-24 09:57:33 -07:00
fcn_r101-d8_512x1024_80k_fp16_cityscapes.py [Bug fix] fixed fp16 inference (#497) 2021-04-24 09:57:33 -07:00
fp16.yml [Enhancement] Add Dev tools to boost develop (#798) 2021-09-02 09:44:51 -07:00
pspnet_r101-d8_512x1024_80k_fp16_cityscapes.py [Bug fix] fixed fp16 inference (#497) 2021-04-24 09:57:33 -07:00

README.md

Mixed Precision Training

Introduction

@article{micikevicius2017mixed,
  title={Mixed precision training},
  author={Micikevicius, Paulius and Narang, Sharan and Alben, Jonah and Diamos, Gregory and Elsen, Erich and Garcia, David and Ginsburg, Boris and Houston, Michael and Kuchaiev, Oleksii and Venkatesh, Ganesh and others},
  journal={arXiv preprint arXiv:1710.03740},
  year={2017}
}

Results and models

Cityscapes

Method Backbone Crop Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) config download
FCN R-101-D8 512x1024 80000 5.37 8.64 76.80 - config model | log
PSPNet R-101-D8 512x1024 80000 5.34 8.77 79.46 - config model | log
DeepLabV3 R-101-D8 512x1024 80000 5.75 3.86 80.48 - config model | log
DeepLabV3+ R-101-D8 512x1024 80000 6.35 7.87 80.46 - config model | log