Update run_dali.sh

2020-11-19 14:58:55 +08:00 · 2020-11-19 14:58:55 +08:00 · d454a22e2c
parent c5b3422b7c
commit d454a22e2c
1 changed files with 2 additions and 0 deletions
--- 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 \