Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSeg
提交
56523aaf
P
PaddleSeg
项目概览
PaddlePaddle
/
PaddleSeg
通知
289
Star
8
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
53
列表
看板
标记
里程碑
合并请求
3
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleSeg
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
53
Issue
53
列表
看板
标记
里程碑
合并请求
3
合并请求
3
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
56523aaf
编写于
6月 29, 2020
作者:
C
chenguowei01
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update train.py and utils.py
上级
4db05742
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
5 addition
and
1 deletion
+5
-1
dygraph/train.py
dygraph/train.py
+1
-1
dygraph/utils/utils.py
dygraph/utils/utils.py
+4
-0
未找到文件。
dygraph/train.py
浏览文件 @
56523aaf
...
@@ -121,7 +121,7 @@ def parse_args():
...
@@ -121,7 +121,7 @@ def parse_args():
parser
.
add_argument
(
parser
.
add_argument
(
'--use_vdl'
,
'--use_vdl'
,
dest
=
'use_vdl'
,
dest
=
'use_vdl'
,
help
=
'Whether to record the data
during training to VisualDL
'
,
help
=
'Whether to record the data
to VisualDL during training
'
,
action
=
'store_true'
)
action
=
'store_true'
)
return
parser
.
parse_args
()
return
parser
.
parse_args
()
...
...
dygraph/utils/utils.py
浏览文件 @
56523aaf
...
@@ -76,6 +76,8 @@ def load_pretrained_model(model, pretrained_model):
...
@@ -76,6 +76,8 @@ def load_pretrained_model(model, pretrained_model):
raise
ValueError
(
raise
ValueError
(
'The pretrained model directory is not Found: {}'
.
formnat
(
'The pretrained model directory is not Found: {}'
.
formnat
(
pretrained_model
))
pretrained_model
))
else
:
logging
.
info
(
'No pretrained model to load, train from scratch'
)
def
resume
(
model
,
optimizer
,
resume_model
):
def
resume
(
model
,
optimizer
,
resume_model
):
...
@@ -94,6 +96,8 @@ def resume(model, optimizer, resume_model):
...
@@ -94,6 +96,8 @@ def resume(model, optimizer, resume_model):
raise
ValueError
(
raise
ValueError
(
'The resume model directory is not Found: {}'
.
formnat
(
'The resume model directory is not Found: {}'
.
formnat
(
resume_model
))
resume_model
))
else
:
logging
.
info
(
'No model need to resume'
)
def
visualize
(
image
,
result
,
save_dir
=
None
,
weight
=
0.6
):
def
visualize
(
image
,
result
,
save_dir
=
None
,
weight
=
0.6
):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录