def test_lstm(node, use_gpu, ignore_bias): node = Variable(node) assert_cuda_active(use_gpu) layer1 = Lstm(output_size=4, ignore_bias=ignore_bias) def func(node): loss = 0 for _ in range(3): loss = sum(layer1(node)) layer1.truncate() return loss compare(func, node, node) for k in layer1.params.keys(): compare(func, layer1.params[k], node)
def test_lstm_temporal_connection(node, use_gpu): node = Variable(node) set_cuda_active(use_gpu) layer1 = Lstm(output_size=4) def func(node): loss = 0 for _ in range(3): loss = sum(layer1(node)) layer1.truncate() return loss compare(func, node, node) for k in layer1.params.keys(): compare(func, layer1.params[k], node)