Open NSMBGhost opened 1 year ago
训练自己的数据报RuntimeError: CUDA error: device-side assert triggered CUDA kernel errors mimight be asynchronously的错误( 已经把DINO_4scale.py中的两个参数修改了)
很奇怪,我的类别是4,num_classes要改成5才不会报以上错误
@NSMBGhost 你的class id是0,1,2,3还是1,2,3,4呢,1,2,3,4的话要设成5
训练自己的数据报RuntimeError: CUDA error: device-side assert triggered CUDA kernel errors mimight be asynchronously的错误( 已经把DINO_4scale.py中的两个参数修改了)