Initializing our weights and biases with random values * 0.01
weights = {
'w1' : tf.Variable(tf.random_normal([n_input_unit, n_hidden_unit_1]) * 0.01),
'w2' : tf.Variable(tf.random_normal([n_hidden_unit_1, n_output_unit]) * 0.01)
}
# logging our weights
tf.summary.histogram("Weight_1__",weights["w1"])
tf.summary.histogram("Weight_2__",weights["w2"])