提交 b85e4841 编写于 作者: D danleifeng

add env params for speeding up in fp16

上级 8904a6e2
...@@ -495,6 +495,12 @@ PLSC支持混合精度训练。使用混合精度训练可以提升训练的速 ...@@ -495,6 +495,12 @@ PLSC支持混合精度训练。使用混合精度训练可以提升训练的速
可以通过下面的代码设置开启混合精度训练: 可以通过下面的代码设置开启混合精度训练:
```python ```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 from plsc import Entry
def main(): def main():
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册