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


Detailed Description

Helper class for constructing residual blocks.

Definition at line 18 of file

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