Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
models
提交
f0ce5cb8
M
models
项目概览
PaddlePaddle
/
models
接近 2 年 前同步成功
通知
230
Star
6828
Fork
2962
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
602
列表
看板
标记
里程碑
合并请求
255
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
models
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
602
Issue
602
列表
看板
标记
里程碑
合并请求
255
合并请求
255
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
f0ce5cb8
编写于
7月 17, 2018
作者:
B
baiyf
提交者:
GitHub
7月 17, 2018
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Temporarily turn off mem_opt and will reopen it later (#1053)
* close mem_opt in pyramidbox
上级
bb59b41a
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
3 addition
and
3 deletion
+3
-3
fluid/face_detection/README_cn.md
fluid/face_detection/README_cn.md
+1
-1
fluid/face_detection/train.py
fluid/face_detection/train.py
+2
-2
未找到文件。
fluid/face_detection/README_cn.md
浏览文件 @
f0ce5cb8
...
@@ -93,7 +93,7 @@ tar -xf vgg_ilsvrc_16_fc_reduced.tar.gz && rm -f vgg_ilsvrc_16_fc_reduced.tar.gz
...
@@ -93,7 +93,7 @@ tar -xf vgg_ilsvrc_16_fc_reduced.tar.gz && rm -f vgg_ilsvrc_16_fc_reduced.tar.gz
`train.py`
是训练模块的主要执行程序,调用示例如下:
`train.py`
是训练模块的主要执行程序,调用示例如下:
```
bash
```
bash
python
-u
train.py
--batch_size
=
1
6
--pretrained_model
=
vgg_ilsvrc_16_fc_reduced
python
-u
train.py
--batch_size
=
1
2
--pretrained_model
=
vgg_ilsvrc_16_fc_reduced
```
```
-
可以通过设置
`export CUDA_VISIBLE_DEVICES=0,1,2,3`
指定想要使用的GPU数量。
-
可以通过设置
`export CUDA_VISIBLE_DEVICES=0,1,2,3`
指定想要使用的GPU数量。
-
更多的可选参数见:
-
更多的可选参数见:
...
...
fluid/face_detection/train.py
浏览文件 @
f0ce5cb8
...
@@ -16,14 +16,14 @@ add_arg = functools.partial(add_arguments, argparser=parser)
...
@@ -16,14 +16,14 @@ add_arg = functools.partial(add_arguments, argparser=parser)
# yapf: disable
# yapf: disable
add_arg
(
'parallel'
,
bool
,
True
,
"Whether use multi-GPU/threads or not."
)
add_arg
(
'parallel'
,
bool
,
True
,
"Whether use multi-GPU/threads or not."
)
add_arg
(
'learning_rate'
,
float
,
0.001
,
"The start learning rate."
)
add_arg
(
'learning_rate'
,
float
,
0.001
,
"The start learning rate."
)
add_arg
(
'batch_size'
,
int
,
1
6
,
"Minibatch size."
)
add_arg
(
'batch_size'
,
int
,
1
2
,
"Minibatch size."
)
add_arg
(
'num_passes'
,
int
,
160
,
"Epoch number."
)
add_arg
(
'num_passes'
,
int
,
160
,
"Epoch number."
)
add_arg
(
'use_gpu'
,
bool
,
True
,
"Whether use GPU."
)
add_arg
(
'use_gpu'
,
bool
,
True
,
"Whether use GPU."
)
add_arg
(
'use_pyramidbox'
,
bool
,
True
,
"Whether use PyramidBox model."
)
add_arg
(
'use_pyramidbox'
,
bool
,
True
,
"Whether use PyramidBox model."
)
add_arg
(
'model_save_dir'
,
str
,
'output'
,
"The path to save model."
)
add_arg
(
'model_save_dir'
,
str
,
'output'
,
"The path to save model."
)
add_arg
(
'resize_h'
,
int
,
640
,
"The resized image height."
)
add_arg
(
'resize_h'
,
int
,
640
,
"The resized image height."
)
add_arg
(
'resize_w'
,
int
,
640
,
"The resized image width."
)
add_arg
(
'resize_w'
,
int
,
640
,
"The resized image width."
)
add_arg
(
'with_mem_opt'
,
bool
,
Tru
e
,
"Whether to use memory optimization or not."
)
add_arg
(
'with_mem_opt'
,
bool
,
Fals
e
,
"Whether to use memory optimization or not."
)
add_arg
(
'pretrained_model'
,
str
,
'./vgg_ilsvrc_16_fc_reduced/'
,
"The init model path."
)
add_arg
(
'pretrained_model'
,
str
,
'./vgg_ilsvrc_16_fc_reduced/'
,
"The init model path."
)
#yapf: enable
#yapf: enable
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录