Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | List of all members
caffe2.python.session.LocalSession Class Reference
Inheritance diagram for caffe2.python.session.LocalSession:
caffe2.python.session.Session

Public Member Functions

def __init__ (self, ws=None)
 
- Public Member Functions inherited from caffe2.python.session.Session
def __init__ (self)
 
def is_open (self)
 
def compile (cls, runnable, workspace_type=None, setup_net_list=None)
 
def run (self, runnable, workspace_type=None, setup_net_list=None)
 
def close (self)
 
def fetch_output (self, output)
 
def __enter__ (self)
 
def __exit__ (self, ex_type, value, traceback)
 

Detailed Description

Session that runs in a single node.
Tasks are all remapped to run in parallel in the 'local' node.

Currently, LocalSession runs all parallel tasks in the same workspace,
but this behavior may change in the future. Only tasks pointing to the
same logical node are guaranteed to always run in the same workspace.

Definition at line 191 of file session.py.


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