Hugging Face's logo

Spaces:
henry000
/
YOLO
Running

App Files Community
YOLO / yolo /tools /solver.py

Commit History

💚 [Pass] format check and refactor code
fa548df

henry000 commited on

🐛 [Fix] validation_step (#134)
8094323
unverified

Adam Kunák commited on

♻️ [Update] pycoco with faster coco, fast validate
3e180a7

henry000 commited on

🔀 [Merge] branch 'TRAIN' into TEST
2ab865c

henry000 commited on

🚑️ [Fix] broadcast of EMA and sync_dist only in val
89a6526

henry000 commited on

🚸 [Add] try-except in loading cache files
1690354

henry000 commited on

✨ [Add] new EMA! for torch lightning
c4cd90a

henry000 commited on

🐛 [Fix] image size order inconsistencies (#118)
959b9b0
unverified

Adam Kunák commited on

✨ [Add] a momentum schedule for wramup epoch
4be6676

henry000 commited on

🐛 [Fix] bugs in dynamic shape in training
3092710

henry000 commited on

✨ [Add] dynamic image size loader
aba5422

henry000 commited on

🐛 [Fix] the bug in lr scheduler, use ceil
8553686

henry000 commited on

🔥 [Remove] stream inference mode
d65babf

henry000 commited on

🐛 [Fix] quite mode bugs, enable to disable :)
3ebbbd9

henry000 commited on

🔀 [Merge] branch 'Lightning' into TEST
30009ba

henry000 commited on

✨ [Add] inference/predict or test mode
7f8235a

henry000 commited on

✏️ [Fix] Process <- Proccess typo
abc3992

henry000 commited on

🐛 [Fix] display on DDP mode
91a3f78

henry000 commited on

🌐 [Add] logging of optimizre(learning rate)
5fcc6be

henry000 commited on

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

henry000 commited on

🐛 [Fix] bug in DDP sync_dist
fc29438

henry000 commited on

🔊 [Update] log item, format value on wandb
240dcb0

henry000 commited on

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

henry000 commited on

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

henry000 commited on

🔨 [FIX] Fixes memory leak (#83)
dea5a8a
unverified

Martin Cerman mcerman-enlite commited on

⬆️ [Update] mAP from 0~1 to percentage
0efc848

henry000 commited on

🐛 [Fix] mul-GPU @ valid bugs, return mAP @ rk 0
a44a5c1

henry000 commited on

✨ [Add] mutli-GPU on validation mode
e0c8580

henry000 commited on

🎨 [Fix] logging bugs, only log on rank0 gpu
ed7d166

henry000 commited on

👽️ [Update] cudaamp, most torch not support amp yet
360e5d7

henry000 commited on

🐛 [Fix] loss, avg loss if using ddp
3b24cc8

henry000 commited on

🚨 [Update] cuda.amp to amp, fix torch warning
9fc0d0e

henry000 commited on

✨ [Add] Visualize image after validation, train
9c191b9

henry000 commited on

🚚 [Update] the class_num&class_list to dataset
3eb85fd

henry000 commited on

🔀 [Merge] branch 'TRAIN' into TEST
ba6baa6

henry000 commited on

✏️ [Fix] typo in loading coco gt files
ce70a60

henry000 commited on

🔨 [Update] AMP, unscale gradient and update model
192b787

henry000 commited on

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

henry000 commited on

✅ [Fix] some path bug and enable ignore run pycoco
b038f54

henry000 commited on

♻️ [Refactor] os.path -> pathlib Path, more safely
fa09d11

henry000 commited on

✨ [Add] saving function when higher mAP
73b88fc

henry000 commited on

✅ [Add] tests, increase test coverage
88e45b9

henry000 commited on

🐛 [Fix] Some display bug, enable print, log
46c9d1c

henry000 commited on

💬 [Update] logging, add avg loss and ap table
c3b133e

henry000 commited on

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

henry000 commited on

✨ [New] inference with coco format
999037c

henry000 commited on

🔀 [Merge] branch 'DEPLOY' into TEST
bfba0a7

henry000 commited on

🔀 [Merge] branch 'INFERENCE' into DEPLOY
2222e19

henry000 commited on

🔀 [Merge] branch 'INFERENCE' into TEST
f5a3a55

henry000 commited on

🚀 [Update] the postproccess -> PostProccess class
329fd0a

henry000 commited on