From d454a22e2c93182ed3b875ef6ced90b948eff91a Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Thu, 19 Nov 2020 14:58:55 +0800
Subject: [PATCH] Update run_dali.sh

---
 tools/run_dali.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tools/run_dali.sh b/tools/run_dali.sh
index 1234f3a8..4cadc346 100755
--- a/tools/run_dali.sh
+++ b/tools/run_dali.sh
@@ -1,5 +1,7 @@
 #!/usr/bin/env bash
 
+export FLAGS_fraction_of_gpu_memory_to_use=0.80
+
 python3.7 -m paddle.distributed.launch \
     --selected_gpus="0,1,2,3" \
     tools/train.py \
-- 
GitLab