shenyunhang / APE

[CVPR 2024] Aligning and Prompting Everything All at Once for Universal Visual Perception
https://arxiv.org/abs/2312.02153
Apache License 2.0
459 stars 28 forks source link

单卡推理 #56

Open wuyefa opened 1 week ago

wuyefa commented 1 week ago

请问改如何将您提供的参数修改为能在单卡上进行推理的版本? 我现在的测试脚本是: python tools/train_net.py \ --eval-only \ --num-gpus 1 \ --resume \ --config-file configs/ape_deta/ape_train.py \ train.output_dir=output/val/odinw_MountainDewCommercial_test \ train.init_checkpoint=output/coco_06_24_v2/model_best.pth 报错显示: AttributeError: 'SomeThing' object has no attribute 'module'