Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleHub
提交
90b45a00
P
PaddleHub
项目概览
PaddlePaddle
/
PaddleHub
1 年多 前同步成功
通知
283
Star
12117
Fork
2091
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
200
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleHub
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
200
Issue
200
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
90b45a00
编写于
11月 22, 2019
作者:
Z
zhangxuefei
提交者:
wuzewu
11月 25, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update config (drop the memory optimization)
上级
f0c2925d
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
9 addition
and
11 deletion
+9
-11
paddlehub/finetune/config.py
paddlehub/finetune/config.py
+8
-4
paddlehub/finetune/task/basic_task.py
paddlehub/finetune/task/basic_task.py
+0
-6
paddlehub/version.py
paddlehub/version.py
+1
-1
未找到文件。
paddlehub/finetune/config.py
浏览文件 @
90b45a00
...
...
@@ -30,14 +30,14 @@ class RunConfig(object):
def
__init__
(
self
,
log_interval
=
10
,
eval_interval
=
100
,
use_pyreader
=
Fals
e
,
use_data_parallel
=
Fals
e
,
use_pyreader
=
Tru
e
,
use_data_parallel
=
Tru
e
,
save_ckpt_interval
=
None
,
use_cuda
=
True
,
checkpoint_dir
=
None
,
num_epoch
=
1
,
batch_size
=
32
,
enable_memory_optim
=
Tru
e
,
enable_memory_optim
=
Fals
e
,
strategy
=
None
):
""" Construct finetune Config """
self
.
_log_interval
=
log_interval
...
...
@@ -53,7 +53,11 @@ class RunConfig(object):
self
.
_strategy
=
DefaultStrategy
()
else
:
self
.
_strategy
=
strategy
self
.
_enable_memory_optim
=
enable_memory_optim
if
enable_memory_optim
:
logger
.
warning
(
"The memory optimization feature has been dropped! PaddleHub now doesn't optimize the memory of the program."
)
self
.
_enable_memory_optim
=
False
if
checkpoint_dir
is
None
:
now
=
int
(
time
.
time
())
time_str
=
time
.
strftime
(
"%Y%m%d%H%M%S"
,
time
.
localtime
(
now
))
...
...
paddlehub/finetune/task/basic_task.py
浏览文件 @
90b45a00
...
...
@@ -134,10 +134,6 @@ class BasicTask(object):
self
.
exe
=
fluid
.
Executor
(
place
=
self
.
place
)
self
.
build_strategy
=
fluid
.
BuildStrategy
()
if
self
.
config
.
enable_memory_optim
:
self
.
build_strategy
.
memory_optimize
=
True
else
:
self
.
build_strategy
.
memory_optimize
=
False
# log item
if
not
os
.
path
.
exists
(
self
.
config
.
checkpoint_dir
):
...
...
@@ -281,8 +277,6 @@ class BasicTask(object):
share_vars_from
=
self
.
_base_compiled_program
if
not
self
.
config
.
use_data_parallel
:
if
self
.
config
.
enable_memory_optim
:
fluid
.
memory_optimize
(
self
.
env
.
main_program
)
self
.
env
.
main_program_compiled
=
None
else
:
self
.
env
.
main_program_compiled
=
fluid
.
CompiledProgram
(
...
...
paddlehub/version.py
浏览文件 @
90b45a00
...
...
@@ -13,5 +13,5 @@
# See the License for the specific language governing permissions and
# limitations under the License.
""" PaddleHub version string """
hub_version
=
"1.
3
.0"
hub_version
=
"1.
4
.0"
module_proto_version
=
"1.0.0"
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录