Module caffe2.python.lstm_benchmark. More...

Functions
def	generate_data (T, shape, num_labels)

def	create_model (args, queue, label_queue, input_shape)

def	Caffe2LSTM (args)

def	Benchmark (args)

def	GetArgumentParser ()

Variables
	log = logging.getLogger("lstm_bench")

def	args = GetArgumentParser().parse_args()

	device

Detailed Description

Module caffe2.python.lstm_benchmark.

Function Documentation

def lstm_benchmark.generate_data	(	T,
		shape,
		num_labels
	)

Fill a queue with input data

Definition at line 22 of file lstm_benchmark.py.

lstm_benchmark.device

Initial value:

1 = core.DeviceOption(

2 caffe2_pb2.CUDA if args.gpu else caffe2_pb2.CPU, 0)

def DeviceOption(device_type, cuda_gpu_id=0, random_seed=None)

Definition: core.py:103

Definition at line 235 of file lstm_benchmark.py.