Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
b4678fd3
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
接近 2 年 前同步成功
通知
116
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看板
提交
b4678fd3
编写于
8月 10, 2021
作者:
W
weishengyu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
revert trainer
上级
98f38fa0
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
4 addition
and
14 deletion
+4
-14
ppcls/engine/trainer.py
ppcls/engine/trainer.py
+4
-14
未找到文件。
ppcls/engine/trainer.py
浏览文件 @
b4678fd3
...
...
@@ -38,7 +38,6 @@ from ppcls.utils.config import print_config
from
ppcls.data
import
build_dataloader
from
ppcls.arch
import
build_model
from
ppcls.arch
import
apply_to_static
from
ppcls.arch.backbone.base.theseus_layer
import
TheseusLayer
from
ppcls.loss
import
build_loss
from
ppcls.metric
import
build_metrics
from
ppcls.optimizer
import
build_optimizer
...
...
@@ -76,11 +75,6 @@ class Trainer(object):
self
.
is_rec
=
False
self
.
model
=
build_model
(
self
.
config
[
"Arch"
])
if
"return_patterns"
in
self
.
config
[
"Arch"
]
and
isinstance
(
self
.
model
,
TheseusLayer
):
self
.
model
.
update_res
(
self
.
config
[
"Arch"
][
"return_patterns"
])
self
.
return_inter
=
True
else
:
self
.
return_inter
=
False
# set @to_static for benchmark, skip this by default.
apply_to_static
(
self
.
config
,
self
.
model
)
...
...
@@ -397,11 +391,11 @@ class Trainer(object):
self
.
model
.
train
()
return
eval_result
def
forward
(
self
,
batch
,
res_dict
=
None
):
def
forward
(
self
,
batch
):
if
not
self
.
is_rec
:
out
=
self
.
model
(
batch
[
0
]
,
res_dict
=
res_dict
)
out
=
self
.
model
(
batch
[
0
])
else
:
out
=
self
.
model
(
batch
[
0
],
batch
[
1
]
,
return_dict
=
res_dict
)
out
=
self
.
model
(
batch
[
0
],
batch
[
1
])
return
out
@
paddle
.
no_grad
()
...
...
@@ -659,11 +653,7 @@ class Trainer(object):
image_file_list
.
append
(
image_file
)
if
len
(
batch_data
)
>=
batch_size
or
idx
==
len
(
image_list
)
-
1
:
batch_tensor
=
paddle
.
to_tensor
(
batch_data
)
if
self
.
return_inter
:
res_dict
=
{}
else
:
res_dict
=
None
out
=
self
.
forward
([
batch_tensor
],
res_dict
)
out
=
self
.
forward
([
batch_tensor
])
if
isinstance
(
out
,
list
):
out
=
out
[
0
]
result
=
postprocess_func
(
out
,
image_file_list
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录