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

Public Member Functions

def __init__ (self, cpp_module)
 
def training (self)
 
def training (self, mode)
 
def __repr__ (self)
 

Public Attributes

 cpp_module
 

Detailed Description

A subclass of ``torch.nn.Module`` that wraps a C++ frontend module and
delegates all access.

Definition at line 49 of file cpp.py.


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