optional para added in run_sgd.py