resnet50_trainer Namespace Reference

Module caffe2.python.examples.resnet50_trainer. More...

Functions
def	AddImageInput (model, reader, batch_size, img_size)
def	AddMomentumParameterUpdate (train_model, LR)
def	RunEpoch (args, epoch, train_model, test_model, total_batch_size, num_shards, expname, explog)
def	Train (args)
def	main ()

Variables
	log = logging.getLogger("resnet50_trainer")

Detailed Description

Module caffe2.python.examples.resnet50_trainer.

Function Documentation

AddImageInput()

def resnet50_trainer.AddImageInput	(		model,
			reader,
			batch_size,
			img_size
	)

Image input operator that loads data from reader and
applies certain transformations to the images.

Definition at line 43 of file resnet50_trainer.py.

AddMomentumParameterUpdate()

def resnet50_trainer.AddMomentumParameterUpdate	(		train_model,
			LR
	)

Add the momentum-SGD update.

Definition at line 63 of file resnet50_trainer.py.

RunEpoch()

def resnet50_trainer.RunEpoch	(		args,
			epoch,
			train_model,
			test_model,
			total_batch_size,
			num_shards,
			expname,
			explog
	)

Run one epoch of the trainer.
TODO: add checkpointing here.

Definition at line 94 of file resnet50_trainer.py.