fixed many bugs, including the train_stddev not being squared, and just launched the experiment with the exact grad E