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