From c4adf74957293810b0f81f2df6cc9be377e4b204 Mon Sep 17 00:00:00 2001
From: zhangting2020 <zhangting_2017@163.com>
Date: Fri, 26 Feb 2021 08:22:21 +0000
Subject: [PATCH] fix oom for batch_size=208

---
 tools/static/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/static/train.py b/tools/static/train.py
index f63f6a9a..973b29d2 100644
--- a/tools/static/train.py
+++ b/tools/static/train.py
@@ -68,7 +68,7 @@ def main(args):
     if 'AMP' in config:
         AMP_RELATED_FLAGS_SETTING = {
             'FLAGS_cudnn_exhaustive_search': 1,
-            'FLAGS_conv_workspace_size_limit': 4000,
+            'FLAGS_conv_workspace_size_limit': 1500,
             'FLAGS_cudnn_batchnorm_spatial_persistent': 1,
             'FLAGS_max_inplace_grad_add': 8,
         }
-- 
GitLab