fix tree model data loader bug (#306)
fix tree model data loader bug (#306)

* fix tree model data loader bug

* fix
bytedance/fedlearner,bytedance/fedlearner,bytedance/fedlearner,bytedance/fedlearner,bytedance/fedlearner,bytedance/fedlearner,bytedance/fedlearner