import tensorflow as tf

                # Embedding + 2 LSTM layers + hidden layers + logits layer
                expected_num_vars = 1 + 2 * 2 * FLAGS.rnn_num_layers + 2 * (
                    FLAGS.cl_num_layers) + 2
                self.assertEqual(len(tf.trainable_variables()), expected_num_vars)

    def testEvalGraph(self):
