Remove buggy and unnecessary code from plot.py
Remove grad_std, grad_mean computations, plot_parameters etc from plot.py. Also, remove anything unnecessary.
Remove grad_std, grad_mean computations, plot_parameters etc from plot.py. Also, remove anything unnecessary.