PaddleClas/tools/static/run_dali.sh

12 lines
300 B
Bash
Raw Normal View History

2020-12-02 22:06:23 +08:00
#!/usr/bin/env bash
2020-12-03 15:48:35 +08:00
export CUDA_VISIBLE_DEVICES="0,1,2,3"
2020-12-02 22:06:23 +08:00
export FLAGS_fraction_of_gpu_memory_to_use=0.80
2020-12-03 15:48:35 +08:00
python3.7 -m paddle.distributed.launch \
--gpus="0,1,2,3" \
2020-12-03 15:48:35 +08:00
tools/static/train.py \
2020-12-02 22:06:23 +08:00
-c ./configs/ResNet/ResNet50.yaml \
-o print_interval=10 \
2020-12-03 15:58:07 +08:00
-o use_dali=True