Hugging Face's logo

Spaces:
henry000
/
YOLO
Running

App Files Community
YOLO / yolo /utils /model_utils.py

Commit History

✨ [Update] back the warm up batch step
604c897

henry000 commited on

🐛 [Update] Momentum of the optimizer
42e909b

henry000 commited on

✨ [New] use lightning framework to training!
8b3b3ef

henry000 commited on

💬 [Update] logging -> loguru, consistency logging
0174b5b

henry000 commited on

✨ [Add] mutli-GPU on validation mode
e0c8580

henry000 commited on

🐛 [Fix] LR schedule bugs, batchidx should increase
10420ed

henry000 commited on

🔨 [Remove] device in loss func, not sure correct
60c4943

henry000 commited on

⚡️ [Update] Optimizer, with more fine LR
5b9da41

henry000 commited on

💚 [Fix] MPS CI bugs, enable using MPS
4056352

henry000 commited on

✅ [Pass] test with fixup choosing device
ecc08c8

henry000 commited on

🚑️ [Fix] get_device typo, not return cpu
3b31306

henry000 commited on

🔨 [Add] Anc2Box conveter, for YOLOv7's output
d5a73bd

henry000 commited on

✅ [Pass] test with lower system requirements
2ee7407

henry000 commited on

🔀 [Merge] branch 'TRAIN' into TEST
1ded329

henry000 commited on

🐛 [Fix] #32 bugs, turn 8400->auto anchor size
7f8ebc5

henry000 commited on

✨ [New] local mAP calculation, AP.5:.95 - mAP.5
21a413f

henry000 commited on

🔨 [Add] the PostProccess class to convert predicts
475302b

henry000 commited on

🚀 [New] DDP mode for training model
1fe2937

henry000 commited on

🚧 [WIP] DDP for model training
7d7e199

henry000 commited on

🔧 [Update] all structure of yaml, more readability
97e9dcb

henry000 commited on

🚚 [Rename] All utils, tools, higher readability
dcceddd

henry000 commited on