From b85e4841dfe9de87d27ce16ff936543ae6f1fdd1 Mon Sep 17 00:00:00 2001 From: danleifeng Date: Wed, 5 Feb 2020 08:38:12 +0000 Subject: [PATCH] add env params for speeding up in fp16 --- README.md | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/README.md b/README.md index 30d2ae2..b66a7bf 100644 --- a/README.md +++ b/README.md @@ -495,6 +495,12 @@ PLSC支持混合精度训练。使用混合精度训练可以提升训练的速 可以通过下面的代码设置开启混合精度训练: ```python +# for speed up +export FLAGS_sync_nccl_allreduce=1 +export FLAGS_cudnn_exhaustive_search=0 +export FLAGS_cudnn_batchnorm_spatial_persistent=1 +export FLAGS_eager_delete_tensor_gb=0 + from plsc import Entry def main(): -- GitLab