run_dali.sh 265 字节
Newer Older
L
littletomatodonkey 已提交
1 2
#!/usr/bin/env bash

L
littletomatodonkey 已提交
3 4
export FLAGS_fraction_of_gpu_memory_to_use=0.80

L
littletomatodonkey 已提交
5 6 7 8 9 10
python3.7 -m paddle.distributed.launch \
    --selected_gpus="0,1,2,3" \
    tools/train.py \
        -c configs/ResNet/ResNet50.yaml \
        -o TRAIN.batch_size=256 \
        -o use_dali=True