geometry tasks with rollout search

[zsworld6]

How to run the geometry tasks with VisuoThink with rollout search

[ekonwang]

Thanks for your issue, you can refer the rollout search implementation in visual navigation module.

Here is simplied algorithm:

       def rollout_func(state):
            (...)

        if tree_search:
            tree_search = (TreeSearch Implementation Class)
    
        for i in range(max_reasoning_depth):
            if tree_search:
                nodes = tree_search.step((init_message, messages), tree_span)
                model_response, messages = tree_search.vote(nodes)
            else:
                model_response, messages = chat_vlm(init_message, messages)

            # agent
            reply_dict = agent.receive(model_response)
            reply = reply_dict['content']
            terminator = reply_dict['terminator']
            if terminator:
                messages.append({"role": "user", "content": reply})
                print(messages[-1])
                break
            init_message = reply