未验证 提交 f7ab1c2d 编写于 作者: littletomatodonkey's avatar littletomatodonkey 提交者: GitHub

Update run_dali.sh

上级 c2054df0
#!/usr/bin/env bash
export CUDA_VISIBLE_DEVICES="0,1,2,3"
export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
export FLAGS_fraction_of_gpu_memory_to_use=0.80
python3.7 -m paddle.distributed.launch \
--gpus="0,1,2,3" \
--gpus="0,1,2,3,4,5,6,7" \
ppcls/static//train.py \
-c ./ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml \
-o Global.use_dali=True
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册