Update run_dali.sh

d454a22e · littletomatodonkey · GitHub · c5b3422b · d454a22e
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

tools/run_dali.sh tools/run_dali.sh +2 -0

未找到文件。
--- a/tools/run_dali.sh
+++ b/tools/run_dali.sh
 #!/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 \