def test_bprop(self):
"""
compare `bprop` results for cpu and gpu backends
"""
r = []
for i in xrange(self.N):
batch_size, x_dim = self.rng.random_integers(3000, size=2)
x = self.rng.rand(batch_size, x_dim).astype(np.float32)
device_id = 0
for nonlinearity in ['sigmoid', 'tanh', 'relu']:
state = self.rng.get_state()
quagga.processor_type = 'gpu'
x_gpu = Connector(Matrix.from_npa(x), device_id)
nonlinearity_block = NonlinearityBlock(x_gpu, nonlinearity)
x_gpu.fprop()
nonlinearity_block.fprop()
_, dL_doutput = nonlinearity_block.output.register_usage(
device_id, device_id)
random_matrix = self.rng.rand(dL_doutput.nrows,
dL_doutput.ncols)
dL_doutput.assign(Context(),
Matrix.from_npa(random_matrix, 'float'))
nonlinearity_block.bprop()
dL_dx_gpu = x_gpu.backward_matrix.to_host()
self.rng.set_state(state)
quagga.processor_type = 'cpu'
x_cpu = Connector(Matrix.from_npa(x), device_id)
nonlinearity_block = NonlinearityBlock(x_cpu, nonlinearity)
x_cpu.fprop()
nonlinearity_block.fprop()
_, dL_doutput = nonlinearity_block.output.register_usage(
device_id, device_id)
random_matrix = self.rng.rand(dL_doutput.nrows,
dL_doutput.ncols)
dL_doutput.assign(Context(),
Matrix.from_npa(random_matrix, 'float'))
nonlinearity_block.bprop()
dL_dx_cpu = x_cpu.backward_matrix.to_host()
r.append(np.allclose(dL_dx_gpu, dL_dx_cpu))
self.assertEqual(sum(r), len(r))
def test_bprop(self):
"""
compare `bprop` results for cpu and gpu backends
"""
r = []
for i in xrange(self.N):
batch_size, x_dim = self.rng.random_integers(3000, size=2)
x = self.rng.rand(batch_size, x_dim).astype(np.float32)
device_id = 0
for nonlinearity in ['sigmoid', 'tanh', 'relu']:
state = self.rng.get_state()
quagga.processor_type = 'gpu'
x_gpu = Connector(Matrix.from_npa(x), device_id)
nonlinearity_block = NonlinearityBlock(x_gpu, nonlinearity)
x_gpu.fprop()
nonlinearity_block.fprop()
_, dL_doutput = nonlinearity_block.output.register_usage(device_id, device_id)
random_matrix = self.rng.rand(dL_doutput.nrows, dL_doutput.ncols)
dL_doutput.assign(Context(), Matrix.from_npa(random_matrix, 'float'))
nonlinearity_block.bprop()
dL_dx_gpu = x_gpu.backward_matrix.to_host()
self.rng.set_state(state)
quagga.processor_type = 'cpu'
x_cpu = Connector(Matrix.from_npa(x), device_id)
nonlinearity_block = NonlinearityBlock(x_cpu, nonlinearity)
x_cpu.fprop()
nonlinearity_block.fprop()
_, dL_doutput = nonlinearity_block.output.register_usage(device_id, device_id)
random_matrix = self.rng.rand(dL_doutput.nrows, dL_doutput.ncols)
dL_doutput.assign(Context(), Matrix.from_npa(random_matrix, 'float'))
nonlinearity_block.bprop()
dL_dx_cpu = x_cpu.backward_matrix.to_host()
r.append(np.allclose(dL_dx_gpu, dL_dx_cpu))
self.assertEqual(sum(r), len(r))
def test_theano_grad(self):
class LogisticRegressionLayer(object):
def __init__(self, W, b):
self.W = theano.shared(value=W)
self.b = theano.shared(value=b[0])
def get_output_expr(self, input_expr):
return T.nnet.sigmoid(T.dot(input_expr, self.W) + self.b)
quagga.processor_type = 'gpu'
r = []
for i in xrange(self.N):
batch_size, x_dim = self.rng.random_integers(3000, size=2)
x = self.rng.rand(batch_size, x_dim).astype(np.float32)
lrdot_W = self.rng.rand(x_dim, 1).astype(np.float32)
lrdot_b = self.rng.rand(1, 1).astype(np.float32)
true_labels = self.rng.randint(2, size=(batch_size,
1)).astype(np.float32)
device_id = 0
for nonlinearity in ['sigmoid', 'tanh', 'relu']:
# Theano model
state = self.rng.get_state()
th_x = T.fmatrix()
th_true_labels = T.fmatrix()
lr_layer = LogisticRegressionLayer(lrdot_W, lrdot_b)
if nonlinearity == 'sigmoid':
f = T.nnet.sigmoid
elif nonlinearity == 'tanh':
f = T.tanh
elif nonlinearity == 'relu':
f = T.nnet.relu
probs = lr_layer.get_output_expr(f(th_x))
loss = T.mean(T.nnet.binary_crossentropy(
probs, th_true_labels))
th_grads = T.grad(loss, wrt=[lr_layer.W, lr_layer.b, th_x])
get_theano_grads = theano.function([th_x, th_true_labels],
th_grads)
th_grads = get_theano_grads(x, true_labels)
# quagga model
self.rng.set_state(state)
x_gpu = Connector(Matrix.from_npa(x), device_id)
true_labels_gpu = Connector(Matrix.from_npa(true_labels))
lrdot_W_gpu = Connector(Matrix.from_npa(lrdot_W), device_id)
lrdot_b_gpu = Connector(Matrix.from_npa(lrdot_b), device_id)
nonlinearity_block = NonlinearityBlock(x_gpu, nonlinearity)
lrdot_block = DotBlock(lrdot_W_gpu, lrdot_b_gpu,
nonlinearity_block.output)
sce_block = SigmoidCeBlock(lrdot_block.output, true_labels_gpu)
x_gpu.fprop()
true_labels_gpu.fprop()
lrdot_W_gpu.fprop()
lrdot_b_gpu.fprop()
nonlinearity_block.fprop()
lrdot_block.fprop()
sce_block.fprop()
sce_block.bprop()
lrdot_block.bprop()
nonlinearity_block.bprop()
q_grads = [
lrdot_W_gpu.backward_matrix.to_host(),
lrdot_b_gpu.backward_matrix.to_host(),
x_gpu.backward_matrix.to_host()
]
for q_grad, th_grad in izip(q_grads, th_grads):
r.append(np.allclose(q_grad, th_grad, atol=1e-5))
self.assertEqual(sum(r), len(r))
def test_theano_grad(self):
class LogisticRegressionLayer(object):
def __init__(self, W, b):
self.W = theano.shared(value=W)
self.b = theano.shared(value=b[0])
def get_output_expr(self, input_expr):
return T.nnet.sigmoid(T.dot(input_expr, self.W) + self.b)
quagga.processor_type = 'gpu'
r = []
for i in xrange(self.N):
batch_size, x_dim = self.rng.random_integers(3000, size=2)
x = self.rng.rand(batch_size, x_dim).astype(np.float32)
lrdot_W = self.rng.rand(x_dim, 1).astype(np.float32)
lrdot_b = self.rng.rand(1, 1).astype(np.float32)
true_labels = self.rng.randint(2, size=(batch_size, 1)).astype(np.float32)
device_id = 0
for nonlinearity in ['sigmoid', 'tanh', 'relu']:
# Theano model
state = self.rng.get_state()
th_x = T.fmatrix()
th_true_labels = T.fmatrix()
lr_layer = LogisticRegressionLayer(lrdot_W, lrdot_b)
if nonlinearity == 'sigmoid':
f = T.nnet.sigmoid
elif nonlinearity == 'tanh':
f = T.tanh
elif nonlinearity == 'relu':
f = T.nnet.relu
probs = lr_layer.get_output_expr(f(th_x))
loss = T.mean(T.nnet.binary_crossentropy(probs, th_true_labels))
th_grads = T.grad(loss, wrt=[lr_layer.W, lr_layer.b, th_x])
get_theano_grads = theano.function([th_x, th_true_labels], th_grads)
th_grads = get_theano_grads(x, true_labels)
# quagga model
self.rng.set_state(state)
x_gpu = Connector(Matrix.from_npa(x), device_id)
true_labels_gpu = Connector(Matrix.from_npa(true_labels))
lrdot_W_gpu = Connector(Matrix.from_npa(lrdot_W), device_id)
lrdot_b_gpu = Connector(Matrix.from_npa(lrdot_b), device_id)
nonlinearity_block = NonlinearityBlock(x_gpu, nonlinearity)
lrdot_block = DotBlock(lrdot_W_gpu, lrdot_b_gpu, nonlinearity_block.output)
sce_block = SigmoidCeBlock(lrdot_block.output, true_labels_gpu)
x_gpu.fprop()
true_labels_gpu.fprop()
lrdot_W_gpu.fprop()
lrdot_b_gpu.fprop()
nonlinearity_block.fprop()
lrdot_block.fprop()
sce_block.fprop()
sce_block.bprop()
lrdot_block.bprop()
nonlinearity_block.bprop()
q_grads = [lrdot_W_gpu.backward_matrix.to_host(),
lrdot_b_gpu.backward_matrix.to_host(),
x_gpu.backward_matrix.to_host()]
for q_grad, th_grad in izip(q_grads, th_grads):
r.append(np.allclose(q_grad, th_grad, atol=1e-5))
self.assertEqual(sum(r), len(r))