AssertionError in runpy.py

Haiyang-W / GiT

[ECCV2024 Oral🔥] Official Implementation of "GiT: Towards Generalist Vision Transformer through Universal Language Interface"

https://arxiv.org/abs/2403.09394

Apache License 2.0

293 stars 12 forks source link

AssertionError in runpy.py #2

Closed dldldlfma closed 6 months ago

dldldlfma commented 6 months ago

I try to use this awesome model like below

but it's not working.

Could i ask you what is the problem?

Thank you!

nnnth commented 6 months ago

Your command line arguments are incorrect. The format is:

bash tools/dist_test.sh configs/GiT/zero-shot/zero_shot_cityscapes_det_base.py ${ckpt_file} ${GPU_NUM} --work-dir ${work_dir}

where ckpt_file is the pre-trained model universal_base.pth, GPU_NUM indicates the number of GPUs to be used, which should be greater than or equal to 1.