* Fix dist demo error

* revert trainer_id

* Enhance bipartite_match_op to support argmax matching after bipartite matching.

* Fix typo error.