setup new optimize_param() in model.py