X = T.tensor4('X') # shape: (batchsize, channels, height, width)
X_r = T.itensor4('X_r')
X_transformed = X_r.dimshuffle(0,2,3,1)
input_layer = WrapperLayer(X.dimshuffle(0,2,3,1)) # input reshaped to (batchsize, height, width,3)
pixel_CNN = pixelConv(
input_layer,
3,
DIM,
Q_LEVELS = Q_LEVELS,
name = model.name + ".pxCNN",
num_layers = 12,
)
model.add_layer(pixel_CNN)
output_probab = Softmax(pixel_CNN).output()
cost = T.nnet.categorical_crossentropy(
output_probab.reshape((-1,output_probab.shape[output_probab.ndim - 1])),
X_r.flatten()
).mean()
# in nats
output_image = sample_from_softmax(output_probab)
model.print_params()
params = model.get_params()
grads = T.grad(cost, wrt=params, disconnected_inputs='warn')
def test_gru(depth, input_dim, hidden_dim):
'''hidden_dim and output_dim are usually same'''
model = Model() # To collect parameters and keep track of layers
X = T.tensor3('X') # input
h0 = T.tensor3('h0') # initial hidden state of recurrent nets
last_layer = WrapperLayer(X)
last_dim = input_dim
for i in range(depth):
gru = GRU(last_dim,
hidden_dim,
last_layer,
name="layer_{}".format(i + 1),
s0=h0[i, :, :])
model.add_layer(gru)
last_layer = gru
last_dim = hidden_dim
params = model.get_params()
print(
"Printing order of params. Important to know as this will help set params for cudnn_rnn"
)
model.print_params()
#list_of_param_values = [p.get_value() for p in params] #list of param values
output = last_layer.output() # output tensor
forward_fun = theano.function([X, h0], output) #forward function
#Y = T.tensor3('Y') # proxy tensor with which we want to match the output of rnn to get a loss
'''For checking gradient, I am defining loss as following,
here 'output' is the theano tensor representing output of rnn op'''
#loss = T.mean((Y - output)*(Y - output)) # mean square error
#grad = T.grad(loss, params) # list of gradient with respect to parameters
#get_grad = theano.function([X, h0, Y], grad) # getting list of gradients
rnnb = dnn.RNNBlock('float32', hidden_dim, depth, 'gru')
psize = rnnb.get_param_size([2, input_dim])
params_cudnn = theano.shared(numpy.zeros((psize, ), dtype='float32'))
# irm, irb, ium, iub, inm, inb, rrm, rrb, rum, rub, rnm, rnb
l0params = rnnb.split_params(params_cudnn, 0, [2, input_dim])
for i, p in enumerate(l0params):
val = params[i].get_value()
p[:] = val
cudnn_rnn_gru_output = rnnb.apply(params_cudnn, X, h0)
#import sys;sys.exit(0)
'''
loss_rnn = T.mean((Y-output_cudnn)*(Y - output_cudnn))
grad_cudnn = T.grad(loss, params_cudnn)
'''
cudnn_rnn_forward_fun = theano.function([X, h0], cudnn_rnn_gru_output)
# h0 = numpy.random.random((1, 2, hidden_dim)).astype('float32')
# inp1 = numpy.random.random((5, 2, input_dim)).astype('float32')
# out = cudnn_rnn_forward_fun(inp1, h0)
# for s in out:
# print(s.shape)
# import sys;sys.exit(0)
def test0(bs, ts):
'''
bs: batch_size
ts: number of timesteps
'''
h0 = numpy.random.random((depth, bs, hidden_dim)).astype('float32')
inp1 = numpy.random.random((bs, ts, input_dim)).astype('float32')
out1 = forward_fun(inp1, h0)
# '''checking output shape'''
assert (out1.shape == (bs, ts, hidden_dim))
hy, y = cudnn_rnn_forward_fun(inp1.transpose((1, 0, 2)), h0)
print(hy.shape, y.shape)
assert (check_equality_two_nd_array(
numpy.asarray(hy)[-1],
numpy.asarray(y)[0]))
print(out1.shape)
print(numpy.asarray(hy).transpose((1, 0, 2)).shape)
assert (check_equality_two_nd_array(out1.transpose((1, 0, 2)),
numpy.asarray(hy)))
sys.exit(0)
def test1(bs, ts):
'''
bs: batch_size
ts: number of timesteps
'''
inp1 = numpy.random.random((bs, ts, input_dim)).astype('float32')
h0 = numpy.random.random((depth, bs, hidden_dim)).astype('float32')
Y = numpy.random.random((bs, ts, hidden_dim)).astype('float32')
grad1 = get_grad(inp1, h0, Y)
'''
grad_cudnn = get_grad_cudnn(inp1, h0, Y)
'''
'''
compare grad with cudnn_grad here
'''
'''
for g, g_hat in zip(grad1, grad_cudnn):
check_equality_two_nd_array(g, g_hat)
'''
test0(2, 5)
print("passed test0 -1")
import sys
sys.exit(0)
test0(1, 10)
print("passed test0 -2")
test1(5, 3)
print("passed test1 -1")
test1(6, 10)
print("passed test1 -2")
X = T.tensor3('X') # shape: (batchsize, height, width)
X_r = T.itensor3('X_r') #shape: (batchsize, height, width)
input_layer = WrapperLayer(X.dimshuffle(0,1,2,'x')) # input reshaped to (batchsize, height, width,1)
pixel_CNN = pixelConv(
input_layer,
1,
DIM,
name = model.name + ".pxCNN",
num_layers = 12,
Q_LEVELS = Q_LEVELS
)
model.add_layer(pixel_CNN)
output_probab = Softmax(pixel_CNN).output()
# in nats
cost = T.nnet.categorical_crossentropy(
output_probab.reshape((-1,output_probab.shape[output_probab.ndim - 1])),
X_r.flatten()
).mean()
output_image = sample_from_softmax(output_probab)
model.print_params()
