robodhruv / visualnav-transformer

Official code and checkpoint release for mobile robot foundation models: GNM, ViNT, and NoMaD.
http://general-navigation-models.github.io
MIT License
425 stars 56 forks source link

Questions about the training #29

Open xzxzmmnn opened 1 month ago

xzxzmmnn commented 1 month ago

Thank you for providing the code. I have successfully trained the model from scratch using the public dataset you shared. The training process seemed to be successful; I've monitored the loss decrease and the visualization results in wandb appear to be satisfactory.

However, when I run the explore.py code using the model checkpoint file from my training, the model outputs are incorrect. To verify the functionality of explore.py, I also executed the code using the model checkpoint file you provided, which worked perfectly fine.

So, I am wondering if I need to modify the explore.py code to better utilize the model that I trained from scratch?

liumingzs commented 2 weeks ago

I want to know how many datasets you download,Can I only download the recon to train the model?

ajaysridhar0 commented 1 day ago

@xzxzmmnn Can you describe how the model outputs are incorrect? @liumingzs You should be able to download all the datasets here. Let me know if you run into any specific issues.