未验证 提交 ab92b397 编写于 作者: C cuicheng01 提交者: GitHub

Merge pull request #1300 from TingquanGao/dev/fix_config

fix: fix config
...@@ -28,7 +28,7 @@ export CUDA_VISIBLE_DEVICES="0" ...@@ -28,7 +28,7 @@ export CUDA_VISIBLE_DEVICES="0"
# 设置用于神经网络训练的显存大小,可根据具体情况设置,一般可设置为0.8或0.7,剩余显存则预留DALI使用 # 设置用于神经网络训练的显存大小,可根据具体情况设置,一般可设置为0.8或0.7,剩余显存则预留DALI使用
export FLAGS_fraction_of_gpu_memory_to_use=0.80 export FLAGS_fraction_of_gpu_memory_to_use=0.80
python tools/static/train.py -c configs/ResNet/ResNet50.yaml -o use_dali=True python ppcls/static/train.py -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml -o use_dali=True
``` ```
也可以使用多卡训练: 也可以使用多卡训练:
...@@ -42,8 +42,8 @@ export FLAGS_fraction_of_gpu_memory_to_use=0.80 ...@@ -42,8 +42,8 @@ export FLAGS_fraction_of_gpu_memory_to_use=0.80
python -m paddle.distributed.launch \ python -m paddle.distributed.launch \
--gpus="0,1,2,3,4,5,6,7" \ --gpus="0,1,2,3,4,5,6,7" \
tools/static/train.py \ ppcls/static/train.py \
-c ./configs/ResNet/ResNet50.yaml \ -c ./ppcls/configs/ImageNet/ResNet/ResNet50.yaml \
-o use_dali=True -o use_dali=True
``` ```
...@@ -56,6 +56,6 @@ export FLAGS_fraction_of_gpu_memory_to_use=0.8 ...@@ -56,6 +56,6 @@ export FLAGS_fraction_of_gpu_memory_to_use=0.8
python -m paddle.distributed.launch \ python -m paddle.distributed.launch \
--gpus="0,1,2,3,4,5,6,7" \ --gpus="0,1,2,3,4,5,6,7" \
tools/static/train.py \ ppcls/static/train.py \
-c configs/ResNet/ResNet50_fp16.yaml -c ./ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml
``` ```
...@@ -17,6 +17,8 @@ Global: ...@@ -17,6 +17,8 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: DeiT_base_distilled_patch16_224 name: DeiT_base_distilled_patch16_224
drop_path_rate : 0.1
drop_rate : 0.0
class_num: 1000 class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
......
...@@ -17,6 +17,8 @@ Global: ...@@ -17,6 +17,8 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: DeiT_base_distilled_patch16_384 name: DeiT_base_distilled_patch16_384
drop_path_rate : 0.1
drop_rate : 0.0
class_num: 1000 class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
......
...@@ -17,6 +17,8 @@ Global: ...@@ -17,6 +17,8 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: DeiT_base_patch16_224 name: DeiT_base_patch16_224
drop_path_rate : 0.1
drop_rate : 0.0
class_num: 1000 class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
......
...@@ -17,6 +17,8 @@ Global: ...@@ -17,6 +17,8 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: DeiT_base_patch16_384 name: DeiT_base_patch16_384
drop_path_rate : 0.1
drop_rate : 0.0
class_num: 1000 class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
......
...@@ -17,6 +17,8 @@ Global: ...@@ -17,6 +17,8 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: DeiT_small_distilled_patch16_224 name: DeiT_small_distilled_patch16_224
drop_path_rate : 0.1
drop_rate : 0.0
class_num: 1000 class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
......
...@@ -17,6 +17,8 @@ Global: ...@@ -17,6 +17,8 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: DeiT_small_patch16_224 name: DeiT_small_patch16_224
drop_path_rate : 0.1
drop_rate : 0.0
class_num: 1000 class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
......
...@@ -17,6 +17,8 @@ Global: ...@@ -17,6 +17,8 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: DeiT_tiny_distilled_patch16_224 name: DeiT_tiny_distilled_patch16_224
drop_path_rate : 0.1
drop_rate : 0.0
class_num: 1000 class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
......
...@@ -17,6 +17,8 @@ Global: ...@@ -17,6 +17,8 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: DeiT_tiny_patch16_224 name: DeiT_tiny_patch16_224
drop_path_rate : 0.1
drop_rate : 0.0
class_num: 1000 class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
......
...@@ -5,7 +5,7 @@ export FLAGS_fraction_of_gpu_memory_to_use=0.80 ...@@ -5,7 +5,7 @@ export FLAGS_fraction_of_gpu_memory_to_use=0.80
python3.7 -m paddle.distributed.launch \ python3.7 -m paddle.distributed.launch \
--gpus="0,1,2,3,4,5,6,7" \ --gpus="0,1,2,3,4,5,6,7" \
ppcls/static//train.py \ ppcls/static/train.py \
-c ./ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml \ -c ./ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml \
-o Global.use_dali=True -o Global.use_dali=True
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册