Which config file do I use to fine-tune the model on the sample datasets?

[frandoLin]

I saw a couple of config files in the repos. Do I use the go1_base_cfg.py for AgiBot G1?

[ModiShi]

It depends on your dataset; you can refer to go1_sft_libero.py to implement a customized configuration for your own collected data.

[frandoLin]

It depends on your dataset; you can refer to go1_sft_libero.py to implement a customized configuration for your own collected data.

That makes sense! The dataset I’m using is the sample dataset from AgiBot World, which I believe was collected using G1. Do I still need to create a custom config file?

[ModiShi]

Yes, you can adapt go1_sft_agilex.py by modifying the SpaceArguments to align with your state and action spaces.