run_dali.sh 216 字节
Newer Older
L
littletomatodonkey 已提交
1 2 3 4 5 6 7 8
#!/usr/bin/env bash

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