From 57710a9df7e17ec0eef5a756065d8339ca9f4096 Mon Sep 17 00:00:00 2001 From: liaoxingyu Date: Wed, 2 Sep 2020 18:07:54 +0800 Subject: [PATCH] add docs for amp training --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index 7a4587f..ff76709 100644 --- a/README.md +++ b/README.md @@ -4,6 +4,7 @@ FastReID is a research platform that implements state-of-the-art re-identificati ## What's New +- [Sep 2020] Automatic Mixed Precision training is supported with pytorch1.6 built-in `torch.cuda.amp`. Set `cfg.SOLVER.AMP_ENABLED=True` to switch it on. - [Aug 2020] [Model Distillation](https://github.com/JDAI-CV/fast-reid/tree/master/projects/DistillReID) is supported, thanks for [guan'an wang](https://github.com/wangguanan)'s contribution. - [Aug 2020] ONNX/TensorRT converter is supported. - [Jul 2020] Distributed training with multiple GPUs, it trains much faster.