Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
BaiXuePrincess
PaddleRec
提交
11f89f9d
P
PaddleRec
项目概览
BaiXuePrincess
/
PaddleRec
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleRec
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleRec
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
11f89f9d
编写于
6月 12, 2020
作者:
C
chengmo
浏览文件
操作
浏览文件
下载
差异文件
Merge remote-tracking branch 'chengmo/fix_ci' into fix_ci
上级
e6443d98
3de2b46f
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
1 addition
and
46 deletion
+1
-46
core/model.py
core/model.py
+1
-3
models/rank/dnn/config.yaml
models/rank/dnn/config.yaml
+0
-43
未找到文件。
core/model.py
浏览文件 @
11f89f9d
...
@@ -138,9 +138,7 @@ class ModelBase(object):
...
@@ -138,9 +138,7 @@ class ModelBase(object):
os
.
environ
[
"FLAGS_communicator_is_sgd_optimizer"
]
=
'0'
os
.
environ
[
"FLAGS_communicator_is_sgd_optimizer"
]
=
'0'
if
name
==
"SGD"
:
if
name
==
"SGD"
:
reg
=
envs
.
get_global_env
(
"hyper_parameters.reg"
,
0.0001
)
optimizer_i
=
fluid
.
optimizer
.
SGD
(
lr
)
optimizer_i
=
fluid
.
optimizer
.
SGD
(
lr
,
regularization
=
fluid
.
regularizer
.
L2DecayRegularizer
(
reg
))
elif
name
==
"ADAM"
:
elif
name
==
"ADAM"
:
optimizer_i
=
fluid
.
optimizer
.
Adam
(
lr
,
lazy_mode
=
True
)
optimizer_i
=
fluid
.
optimizer
.
Adam
(
lr
,
lazy_mode
=
True
)
elif
name
==
"ADAGRAD"
:
elif
name
==
"ADAGRAD"
:
...
...
models/rank/dnn/config.yaml
浏览文件 @
11f89f9d
...
@@ -71,22 +71,6 @@ runner:
...
@@ -71,22 +71,6 @@ runner:
print_interval
:
10
print_interval
:
10
phases
:
[
phase1
]
phases
:
[
phase1
]
-
name
:
single_gpu_train
class
:
train
# num of epochs
epochs
:
4
# device to run training or infer
device
:
gpu
selected_gpus
:
"
2"
save_checkpoint_interval
:
2
# save model interval of epochs
save_inference_interval
:
4
# save inference
save_checkpoint_path
:
"
increment"
# save checkpoint path
save_inference_path
:
"
inference"
# save inference path
save_inference_feed_varnames
:
[]
# feed vars of save inference
save_inference_fetch_varnames
:
[]
# fetch vars of save inference
init_model_path
:
"
"
# load model path
print_interval
:
10
-
name
:
single_cpu_infer
-
name
:
single_cpu_infer
class
:
infer
class
:
infer
# num of epochs
# num of epochs
...
@@ -96,33 +80,6 @@ runner:
...
@@ -96,33 +80,6 @@ runner:
init_model_path
:
"
increment/0"
# load model path
init_model_path
:
"
increment/0"
# load model path
phases
:
[
phase2
]
phases
:
[
phase2
]
-
name
:
local_cluster_cpu_ps_train
class
:
local_cluster
epochs
:
4
device
:
cpu
save_checkpoint_interval
:
2
# save model interval of epochs
save_inference_interval
:
4
# save inference
save_checkpoint_path
:
"
increment"
# save checkpoint path
save_inference_path
:
"
inference"
# save inference path
save_inference_feed_varnames
:
[]
# feed vars of save inference
save_inference_fetch_varnames
:
[]
# fetch vars of save inference
init_model_path
:
"
"
# load model path
print_interval
:
1
-
name
:
multi_gpu_train
class
:
train
epochs
:
4
device
:
gpu
selected_gpus
:
"
2,3"
save_checkpoint_interval
:
2
# save model interval of epochs
save_inference_interval
:
4
# save inference
save_checkpoint_path
:
"
increment"
# save checkpoint path
save_inference_path
:
"
inference"
# save inference path
save_inference_feed_varnames
:
[]
# feed vars of save inference
save_inference_fetch_varnames
:
[]
# fetch vars of save inference
init_model_path
:
"
"
# load model path
print_interval
:
10
# runner will run all the phase in each epoch
# runner will run all the phase in each epoch
phase
:
phase
:
-
name
:
phase1
-
name
:
phase1
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录