Reduced the large memory requirements of ml_trainproximal.