SudeepDasari / data4robotics

MIT License
56 stars 9 forks source link

Fix the issue of download script always trying to download #2

Closed younggyoseo closed 11 months ago

younggyoseo commented 11 months ago

This PR fixes the issue of download script always trying to download the pre-trained weights in an active python workspace directory (e.g., when you want to load the pre-trained parameters in outside project directory via load_vit or load_resnet18)

Congrats on the nice work and hope this helps.

SudeepDasari commented 11 months ago

Thanks for catching this bug, and for checking out the project!

I just pushed a slightly simpler fix (commit link) to the main branch. Let me know if any issues still remain.