Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
caffe2.python.models.resnet.ResNetBuilder Class Reference

Public Member Functions

def __init__ (self, model, prev_blob, no_bias, is_test, bn_epsilon=1e-5, bn_momentum=0.9)
 
def add_conv (self, in_filters, out_filters, kernel, stride=1, group=1, pad=0)
 
def add_relu (self)
 
def add_spatial_bn (self, num_filters)
 
def add_bottleneck (self, input_filters, base_filters, output_filters, stride=1, group=1, spatial_batch_norm=True)
 
def add_simple_block (self, input_filters, num_filters, down_sampling=False, spatial_batch_norm=True)
 

Public Attributes

 model
 
 comp_count
 
 comp_idx
 
 prev_blob
 
 is_test
 
 bn_epsilon
 
 bn_momentum
 
 no_bias
 

Detailed Description

Helper class for constructing residual blocks.

Definition at line 18 of file resnet.py.


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