caffe2.python.session.LocalSession Class Reference

Inheritance diagram for caffe2.python.session.LocalSession:

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 176 of file session.py.

---

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

• caffe2/python/session.py