macrobomaster / cv-yolov7

Custom yolov7 model with opencv
GNU General Public License v3.0
5 stars 3 forks source link

Update loss.py #3

Closed JustinLinKK closed 1 year ago

JustinLinKK commented 1 year ago

Fixed the issue that after the GPU-only updated the main branch can't uses for CPU training