lass/lr/ridge/: fix failing convergence due to memory limitation of yarn container