lovelyqian / CDFSOD-benchmark

A benchmark for cross-domain few-shot object detection (ECCV24 paper: Cross-Domain Few-Shot Object Detection via Enhanced Open-Set Object Detector)
Apache License 2.0
57 stars 6 forks source link

Loss became infinite or NaN #3

Closed NHeLv1 closed 3 months ago

NHeLv1 commented 3 months ago

I ran it on uodd 10-shot and the loss crashed ![Uploading image.png…]()

NHeLv1 commented 3 months ago

image

lovelyqian commented 3 months ago

image

Hi! Thanks for your question, but we tried to reproduce it by three different co-authors, and we didn't cause this error (I actually have no idea about it, may you check maybe the version of your environment.

NHeLv1 commented 3 months ago

thx, I'll try to figure it out

NHeLv1 commented 3 months ago

I solved it by changing batch size to 8, I think there is still a potential problem, but it shouldn't be related to this work

lovelyqian commented 3 months ago

Still thanks for your feedback!