diff --git a/README.md b/README.md index 30d2ae20052c2f084d43128a974cad1d76c0fc84..b66a7bf4ceb0eab9b17e12e15653c3de8c7a1111 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():