Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
662a3909
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
662a3909
编写于
7月 06, 2020
作者:
S
shippingwang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix vd
上级
be87be27
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
55 addition
and
2 deletion
+55
-2
configs/high_performance/dali_mix.yaml
configs/high_performance/dali_mix.yaml
+45
-0
tools/dali.py
tools/dali.py
+1
-0
tools/run_dali.sh
tools/run_dali.sh
+9
-2
未找到文件。
configs/high_performance/dali_mix.yaml
0 → 100644
浏览文件 @
662a3909
mode
:
'
train'
ARCHITECTURE
:
name
:
'
ResNet50_vd'
pretrained_model
:
"
"
model_save_dir
:
"
./output/"
classes_num
:
1000
total_images
:
1281167
save_interval
:
1
validate
:
True
valid_interval
:
1
epochs
:
200
topk
:
5
image_shape
:
[
3
,
224
,
224
]
use_dali
:
True
use_mix
:
True
ls_epsilon
:
0.1
LEARNING_RATE
:
function
:
'
Cosine'
params
:
lr
:
0.1
OPTIMIZER
:
function
:
'
Momentum'
params
:
momentum
:
0.9
regularizer
:
function
:
'
L2'
factor
:
0.000070
TRAIN
:
batch_size
:
256
num_workers
:
4
file_list
:
"
./dataset/ILSVRC2012/train_list.txt"
data_dir
:
"
./dataset/ILSVRC2012/"
shuffle_seed
:
0
VALID
:
batch_size
:
64
num_workers
:
4
file_list
:
"
./dataset/ILSVRC2012/val_list.txt"
data_dir
:
"
./dataset/ILSVRC2012/"
shuffle_seed
:
0
tools/dali.py
浏览文件 @
662a3909
...
@@ -313,6 +313,7 @@ def normalize(feeds, config):
...
@@ -313,6 +313,7 @@ def normalize(feeds, config):
def
mix
(
feeds
,
config
,
is_train
=
True
):
def
mix
(
feeds
,
config
,
is_train
=
True
):
env
=
os
.
environ
gpu_num
=
paddle
.
fluid
.
core
.
get_cuda_device_count
()
if
(
gpu_num
=
paddle
.
fluid
.
core
.
get_cuda_device_count
()
if
(
'PADDLE_TRAINERS_NUM'
)
and
(
'PADDLE_TRAINERS_NUM'
)
and
(
'PADDLE_TRAINER_ID'
'PADDLE_TRAINER_ID'
...
...
tools/run_dali.sh
浏览文件 @
662a3909
...
@@ -5,8 +5,15 @@ export FLAGS_fraction_of_gpu_memory_to_use=0.8
...
@@ -5,8 +5,15 @@ export FLAGS_fraction_of_gpu_memory_to_use=0.8
python3
-m
paddle.distributed.launch
\
python3
-m
paddle.distributed.launch
\
--selected_gpus
=
"0,1"
\
--selected_gpus
=
"0,1
,2,3
"
\
tools/train.py
\
tools/train.py
\
-c
./configs/high_performance/dali.yaml
\
-c
./configs/high_performance/dali
_mix
.yaml
\
-o
use_mix
=
True
\
-o
use_mix
=
True
\
-o
use_dali
=
True
-o
use_dali
=
True
#python3 -m paddle.distributed.launch \
# --selected_gpus="0,1,2,3" \
# tools/train.py \
# -c ./configs/high_performance/dali.yaml \
# -o use_mix=True \
# -o use_dali=True
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录