27 Commits (03f56791a3165bb183d1065937eecb4c023a7195)

Author SHA1 Message Date
Ayush Chaurasia 216cf2ddb6
Release 8.0.4 fixes (#256)
2 years ago
Glenn Jocher 422c49d439
Cleanup (#168)
2 years ago
Glenn Jocher 3cbf3ec455
Global settings typechecking (#148)
2 years ago
Glenn Jocher 99275814f1
Add AutoBatch from YOLOv5 (#145)
2 years ago
Glenn Jocher c5c86a3acd
CLI DDP fixes (#135)
2 years ago
Glenn Jocher 8f3cd52844
Fix `yolo mode=train` CLI bug on model load (#133)
2 years ago
Glenn Jocher af6e3c536b
Override fixes and general updates (#129)
2 years ago
Glenn Jocher c9f3e469cb
Add CoreML iOS updates (#121)
2 years ago
Glenn Jocher 92dad1c1b5
Start export implementation (#110)
2 years ago
Glenn Jocher 6f0ba81427
Add best.pt val and COCO pycocotools val (#98)
2 years ago
Ayush Chaurasia a1808eeda4
logger updates (#97)
2 years ago
Glenn Jocher 98815d560f
YOLOv5 updates (#90)
2 years ago
Glenn Jocher 6432afc5f9
Rename `img_size` to `imgsz` (#86)
2 years ago
Ayush Chaurasia ae2443c210
Add flops, num_params, inference speed logging and best.pt logging (#84)
2 years ago
Ayush Chaurasia e6737f1207
Predictor support (#65)
2 years ago
Ayush Chaurasia 5a52e7663a
standalone val (#56)
2 years ago
Laughing 3a241e4cea
update segment training (#57)
2 years ago
Ayush Chaurasia c5f5b80c04
Deterministic training (#53)
2 years ago
ayush chaurasia 793dde365d Revert "update"
2 years ago
ayush chaurasia 486bc49414 update
2 years ago
Ayush Chaurasia d7df1770fa
Add warmup and accumulation (#52)
2 years ago
Ayush Chaurasia 4291b9c31c
Add EMA and model checkpointing (#49)
2 years ago
Glenn Jocher 27d6545117
General console printout updates (#48)
2 years ago
Ayush Chaurasia f56c9bcc26
Segmentation support & other enchancements (#40)
2 years ago
Ayush Chaurasia 1054819a59
Add initial model interface (#30)
2 years ago
Ayush Chaurasia 7b560f7861
Model builder (#29)
2 years ago
Ayush Chaurasia d0b3c9812b
Trainer + Dataloaders (#27)
2 years ago