changed learning rate in variational_lookahead. now just running variational for 1 iteration
.