[FIX] Pre-initialize all trees before dispatching

This should help fixing #2721 by pre-initializing all trees before dispatching and hopefully avoid the race condition in `get_params`. 

CC: @ogrisel