search

Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
MIT License
1.03k stars 115 forks source link

使用ViT-L训练时出错！ #117

Open RedstoneWill opened 8 months ago

RedstoneWill commented 8 months ago

尊敬的作者大大您好，使用ViT-L训练模型，安装README将vit.py和sfp.py配置好之后，使用单卡训练出现下面的错误： KeyError: "CoDETR: 'ViT is not in the models registry'"

若使用分布式训练则出现下面的错误： TypeError: init_process_group() got an unexpected keyword argument 'port'

请问是什么原因呢？感谢指导！

DoctorDream commented 8 months ago

init.py中需要增加两处：

  1. from .vit import ViT
  2. 在all的list中加上'ViT'
RedstoneWill commented 8 months ago

在init.py中都添加了，根据readme做了，但是不是缺少某些python库或者需要重新编译什么的？

TempleX98 commented 8 months ago

应该是单卡训练的config里指定了ddp的port，可以看下是不是这个问题

Baboom-l commented 7 months ago

检查一下你用的是mmdet2还是mmdet3

lilero commented 5 months ago

应该是单卡训练的config里指定了ddp的port，可以看下是不是这个问题

你好，我发邮件给你了，那个VIT-L模型和配置文件可以发我吗

onnx20 commented 4 months ago

应该是单卡训练的config里指定了ddp的port，可以看下是不是这个问题

你好，我邮件申请使用你们release的ViT-L模型权重和配置文件，用来训练coco数据集失败，我看README里面备注的是[config (test)]，是不是只能测试，不能训练呢，因为我发现co_dino_5scale_vit_large_lvis.py配置文件貌似不完整，缺少optimizer_config等参数配置，我自己添加optimizer_config之后，又出现这个错误assert 'mix_results' in results

TempleX98 commented 1 month ago

@onnx20 现在的code可以吗，我更推荐使用COCO的config训练