zd11024 / NaviLLM

[CVPR 2024] The code for paper 'Towards Learning a Generalist Model for Embodied Navigation'
MIT License
102 stars 7 forks source link

Autoload Datasets & Decouple the Prompt from Model #4

Closed zd11024 closed 9 months ago

zd11024 commented 9 months ago

In this PR, I have made several modifications to make the codebase more clear and flexible.

  1. Implement metaclasses to autoload datasets for different tasks.
  2. Decouple the prompt construction from the models and move it as part of the agent (at `tasks/agents').
  3. Remove unuseful parameters in the model
  4. Fix the bug that causes `pipe broken' when evaluating Meteor.

Please review these changes and provide any feedback or questions you might have. Many Thanks!