Inference only has results on cuda0

IDEA-Research / DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"

Apache License 2.0

2.15k stars 232 forks source link

Open tobymu opened 1 year ago

tobymu commented 1 year ago

@HaoZhang534 发现一个比较奇怪的现象，用自己的数据训好的dino模型在cuda0上能检测到box，换到cuda1~7上就无法检测到box，可能是什么原因呢？

HaoZhang534 commented 1 year ago

有详细描述吗