weights = tf.Variable(

    tf.truncated_normal([IMAGE_PIXELS, hidden1_units],

                        stddev=1.0 / math.sqrt(float(IMAGE_PIXELS))),

    name='weights')

biases = tf.Variable(tf.zeros([hidden1_units]),

                     name='biases')