Trying to use plot_layers to propagate all the way fwd and back to see if greedy training is working