resnet Namespace Reference

Module caffe2.python.models.resnet. More...

Classes
class	ResNetBuilder

Functions
def	create_resnet50 (model, data, num_input_channels, num_labels, label=None, is_test=False, no_loss=False, no_bias=0, conv1_kernel=7, conv1_stride=2, final_avg_kernel=7)
def	create_resnet_32x32 (model, data, num_input_channels, num_groups, num_labels, is_test=False)

Detailed Description

Module caffe2.python.models.resnet.

Function Documentation

create_resnet_32x32()

def resnet.create_resnet_32x32	(		model,
			data,
			num_input_channels,
			num_groups,
			num_labels,
			is_test = False
	)

Create residual net for smaller images (sec 4.2 of He et. al (2015))
num_groups = 'n' in the paper

Definition at line 275 of file resnet.py.