Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleX
提交
fb0dd18a
P
PaddleX
项目概览
PaddlePaddle
/
PaddleX
通知
136
Star
4
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
43
列表
看板
标记
里程碑
合并请求
5
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleX
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
43
Issue
43
列表
看板
标记
里程碑
合并请求
5
合并请求
5
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
fb0dd18a
编写于
5月 12, 2020
作者:
F
FlyingQianMM
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix errors
上级
33aff574
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
4 addition
and
6 deletion
+4
-6
paddlex/cv/models/faster_rcnn.py
paddlex/cv/models/faster_rcnn.py
+1
-2
paddlex/cv/models/mask_rcnn.py
paddlex/cv/models/mask_rcnn.py
+1
-2
paddlex/utils/utils.py
paddlex/utils/utils.py
+2
-2
未找到文件。
paddlex/cv/models/faster_rcnn.py
浏览文件 @
fb0dd18a
...
...
@@ -232,9 +232,8 @@ class FasterRCNN(BaseAPI):
self
.
net_initialize
(
startup_prog
=
fluid
.
default_startup_program
(),
pretrain_weights
=
pretrain_weights
,
fuse_bn
=
fuse_bn
,
save_dir
=
save_dir
,
sensitivities_file
=
sensitivities_file
,
eval_metric_loss
=
eval_metric_loss
,
resume_checkpoint
=
resume_checkpoint
)
start_epoch
=
0
if
resume_checkpoint
:
...
...
paddlex/cv/models/mask_rcnn.py
浏览文件 @
fb0dd18a
...
...
@@ -199,9 +199,8 @@ class MaskRCNN(FasterRCNN):
self
.
net_initialize
(
startup_prog
=
fluid
.
default_startup_program
(),
pretrain_weights
=
pretrain_weights
,
fuse_bn
=
fuse_bn
,
save_dir
=
save_dir
,
sensitivities_file
=
sensitivities_file
,
eval_metric_loss
=
eval_metric_loss
,
resume_checkpoint
=
resume_checkpoint
)
start_epoch
=
0
if
resume_checkpoint
:
...
...
paddlex/utils/utils.py
浏览文件 @
fb0dd18a
...
...
@@ -300,8 +300,8 @@ def load_pretrain_weights(exe,
if
pretrained_shape
!=
actual_shape
:
raise
Exception
(
"Shape of optimizer variable {} doesn't match.(Last: {}, Now: {}), {}"
.
format
(
var
.
name
,
opt_dict
[
var
.
name
].
shape
,
var
.
shape
),
exception_message
)
.
format
(
var
.
name
,
pretrained_
shape
,
actual_
shape
),
exception_message
)
optimizer_varname_list
=
[
var
.
name
for
var
in
optimizer_var_list
]
if
os
.
exists
(
osp
.
join
(
weights_dir
,
'learning_rate'
)
)
and
'learning_rate'
not
in
optimizer_varname_list
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录