frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-07 01:50:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Set up run_every_ms when constructing ExecutionStep

Browse Source 
Summary: same as title.

Differential Revision: D5709274

fbshipit-source-id: f88b1325f3e6b948b836cc90f4d9c38a27be28ab
This commit is contained in:
Lei Chen
2017-08-25 15:46:07 -07:00
committed by Facebook Github Bot
parent ae0c4c8e66
commit 432cba6c05
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
caffe2/python/core.py
View File

@ -2485,6 +2485,8 @@ class ExecutionStep(object):
step_proto.HasField('num_concurrent_instances') else None
create_workspace = step_proto.create_workspace if\
step_proto.HasField('create_workspace') else None
run_every_ms = step_proto.run_every_ms if\
step_proto.HasField('run_every_ms') else None
return execution_step(
step_proto.name,
@ -2496,7 +2498,8 @@ class ExecutionStep(object):
should_stop_blob=should_stop_blob,
only_once=only_once,
num_concurrent_instances=num_concurrent_instances,
create_workspace=create_workspace)
create_workspace=create_workspace,
run_every_ms=run_every_ms)
def add_nets_in_order(step, net_list):
@ -2612,7 +2615,8 @@ def execution_step(default_name,
should_stop_blob=None,
only_once=None,
num_concurrent_instances=None,
create_workspace=False):
create_workspace=False,
run_every_ms=None):
"""
Helper for creating an ExecutionStep.
- steps_or_nets can be:
@ -2648,6 +2652,8 @@ def execution_step(default_name,
step.SetNumConcurrentInstances(num_concurrent_instances)
if create_workspace:
step.SetCreateWorkspace(True)
if run_every_ms:
step.RunEveryMillis(run_every_ms)
if isinstance(steps_or_nets, ExecutionStep):
step.AddSubstep(steps_or_nets)