Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
task.Task Class Reference
Inheritance diagram for task.Task:

Public Member Functions

def __init__ (self, step=None, outputs=None, workspace_type=None, group=None, node=None, name=None)
 
def __enter__ (self)
 
def __exit__ (self, type, value, traceback)
 
def workspace_type (self)
 
def add_output (self, output)
 
def add_outputs (self, outputs)
 
def set_step (self, step)
 
def get_step (self)
 
def output_list (self)
 
def outputs (self)
 

Public Attributes

 node
 
 group
 
 name
 

Static Public Attributes

string TASK_SETUP = 'task_setup'
 
string REPORT_STEP = 'report_step'
 

Detailed Description

A Task is composed of an execution step and zero or more outputs.
Tasks are executed in the context of a TaskGroup, which, in turn, can
be run by a Session.

Task outputs are fetched by the session at the end of the run.

Definition at line 408 of file task.py.

Constructor & Destructor Documentation

◆ __init__()

def task.Task.__init__ (   self,
  step = None,
  outputs = None,
  workspace_type = None,
  group = None,
  node = None,
  name = None 
)
Instantiate a Task and add it to the current TaskGroup and Node.

Definition at line 437 of file task.py.


The documentation for this class was generated from the following file: