run_dali.sh 287 字节
Newer Older
S
shippingwang 已提交
1 2
#!/usr/bin/env bash

S
fix  
shippingwang 已提交
3
export PYTHONPATH=$PWD:$PYTHONPATH
S
shippingwang 已提交
4 5 6 7 8 9 10 11 12
export FLAGS_fraction_of_gpu_memory_to_use=0.8


python3 -m paddle.distributed.launch \
    --selected_gpus="0,1" \
    tools/train.py \
        -c ./configs/high_performance/dali.yaml \
	-o use_mix=True \
        -o use_dali=True