Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
ab92b397
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
大约 1 年 前同步成功
通知
115
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
ab92b397
编写于
10月 15, 2021
作者:
C
cuicheng01
提交者:
GitHub
10月 15, 2021
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #1300 from TingquanGao/dev/fix_config
fix: fix config
上级
70baba54
8a5624f8
变更
10
隐藏空白更改
内联
并排
Showing
10 changed file
with
22 addition
and
6 deletion
+22
-6
docs/zh_CN/extension/train_with_DALI.md
docs/zh_CN/extension/train_with_DALI.md
+5
-5
ppcls/configs/ImageNet/DeiT/DeiT_base_distilled_patch16_224.yaml
...onfigs/ImageNet/DeiT/DeiT_base_distilled_patch16_224.yaml
+2
-0
ppcls/configs/ImageNet/DeiT/DeiT_base_distilled_patch16_384.yaml
...onfigs/ImageNet/DeiT/DeiT_base_distilled_patch16_384.yaml
+2
-0
ppcls/configs/ImageNet/DeiT/DeiT_base_patch16_224.yaml
ppcls/configs/ImageNet/DeiT/DeiT_base_patch16_224.yaml
+2
-0
ppcls/configs/ImageNet/DeiT/DeiT_base_patch16_384.yaml
ppcls/configs/ImageNet/DeiT/DeiT_base_patch16_384.yaml
+2
-0
ppcls/configs/ImageNet/DeiT/DeiT_small_distilled_patch16_224.yaml
...nfigs/ImageNet/DeiT/DeiT_small_distilled_patch16_224.yaml
+2
-0
ppcls/configs/ImageNet/DeiT/DeiT_small_patch16_224.yaml
ppcls/configs/ImageNet/DeiT/DeiT_small_patch16_224.yaml
+2
-0
ppcls/configs/ImageNet/DeiT/DeiT_tiny_distilled_patch16_224.yaml
...onfigs/ImageNet/DeiT/DeiT_tiny_distilled_patch16_224.yaml
+2
-0
ppcls/configs/ImageNet/DeiT/DeiT_tiny_patch16_224.yaml
ppcls/configs/ImageNet/DeiT/DeiT_tiny_patch16_224.yaml
+2
-0
ppcls/static/run_dali.sh
ppcls/static/run_dali.sh
+1
-1
未找到文件。
docs/zh_CN/extension/train_with_DALI.md
浏览文件 @
ab92b397
...
...
@@ -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"
\
too
ls/static/train.py
\
-c
./
configs
/ResNet/ResNet50.yaml
\
ppc
ls/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"
\
too
ls/static/train.py
\
-c
configs
/ResNet/ResNet50_fp16.yaml
ppc
ls/static/train.py
\
-c
./ppcls/configs/ImageNet
/ResNet/ResNet50_fp16.yaml
```
ppcls/configs/ImageNet/DeiT/DeiT_base_distilled_patch16_224.yaml
浏览文件 @
ab92b397
...
...
@@ -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
...
...
ppcls/configs/ImageNet/DeiT/DeiT_base_distilled_patch16_384.yaml
浏览文件 @
ab92b397
...
...
@@ -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
...
...
ppcls/configs/ImageNet/DeiT/DeiT_base_patch16_224.yaml
浏览文件 @
ab92b397
...
...
@@ -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
...
...
ppcls/configs/ImageNet/DeiT/DeiT_base_patch16_384.yaml
浏览文件 @
ab92b397
...
...
@@ -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
...
...
ppcls/configs/ImageNet/DeiT/DeiT_small_distilled_patch16_224.yaml
浏览文件 @
ab92b397
...
...
@@ -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
...
...
ppcls/configs/ImageNet/DeiT/DeiT_small_patch16_224.yaml
浏览文件 @
ab92b397
...
...
@@ -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
...
...
ppcls/configs/ImageNet/DeiT/DeiT_tiny_distilled_patch16_224.yaml
浏览文件 @
ab92b397
...
...
@@ -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
...
...
ppcls/configs/ImageNet/DeiT/DeiT_tiny_patch16_224.yaml
浏览文件 @
ab92b397
...
...
@@ -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
...
...
ppcls/static/run_dali.sh
浏览文件 @
ab92b397
...
...
@@ -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.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录