pretraining.sh launch script is hardcoded to dev folders/env...can you provide a generic launch script for regular servers?

hustvl / Vim

[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model

Apache License 2.0

2.56k stars 160 forks source link

pretraining.sh launch script is hardcoded to dev folders/env...can you provide a generic launch script for regular servers? #5

Closed lessw2020 closed 5 months ago

lessw2020 commented 5 months ago

Can you provide a generic launch script for pre-training with vim? The readme suggests to run: bash vim/scripts/pt-vim-t.sh

but the contents of pt-vimt-t.sh have a bunch of local hardcoded paths that do not exist on regular servers ala: --data-path /share/project/lianghuizhu/datasets/IN1K source /opt/conda/bin/activate /home/zhulianghui/.conda/envs/deit-dev etc.

Thanks very much.

Unrealluver commented 5 months ago

Greetings!

Thanks for your suggestion! We have fixed it.

lessw2020 commented 5 months ago

Thanks very much for the fast fix!