A problem when training customer data

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 NSMBGhost opened 1 year ago

NSMBGhost commented 1 year ago

训练自己的数据报RuntimeError: CUDA error: device-side assert triggered CUDA kernel errors mimight be asynchronously的错误（已经把DINO_4scale.py中的两个参数修改了）

NSMBGhost commented 1 year ago

很奇怪，我的类别是4，num_classes要改成5才不会报以上错误

HaoZhang534 commented 1 year ago

@NSMBGhost 你的class id是0,1,2,3还是1,2,3,4呢，1,2,3,4的话要设成5