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 2 years ago

JustinLinKK commented 2 years ago

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