Add the comments of start()
Fix arguments of _get_module_io_param_map()
Fix arguments of _get_module_io_param_map()
Fix arguments of get_ml_client()