tgxs002 / CORA

A DETR-style framework for open-vocabulary detection (OVD). CVPR 2023
Apache License 2.0
166 stars 14 forks source link

ZeroDivisionError: float division by zero #4

Closed eternaldolphin closed 1 year ago

eternaldolphin commented 1 year ago

image It seems that the code lacks some data filtration?

tgxs002 commented 1 year ago

You should be able to fix it with larger batch size.

eternaldolphin commented 1 year ago

Thank you (and voice out poverty

davidyang180 commented 1 year ago

You should be able to fix it with larger batch size.

If the memory is limited, does adding a very small number (float("1e-8")) compensation to the denominator part affect the model training?