21 Commits (98815d560f11f51ab6244e6cd3151e11aeff9650)

Author SHA1 Message Date
Glenn Jocher 98815d560f
YOLOv5 updates (#90)
2 years ago
Glenn Jocher cb4f20f3cf
Add TensorBoard support (#87)
2 years ago
Glenn Jocher 248d54ca03
Update metrics names (#85)
2 years ago
Glenn Jocher 6432afc5f9
Rename `img_size` to `imgsz` (#86)
2 years ago
Ayush Chaurasia 681cfc1c35
Make config overrides user friendly (#80)
2 years ago
Laughing e629335f6d
Revert augment_hyps (#70)
2 years ago
Ayush Chaurasia e6737f1207
Predictor support (#65)
2 years ago
Laughing fbeeb5d1e1
add resuming (#63)
2 years ago
Ayush Chaurasia 5a52e7663a
standalone val (#56)
2 years ago
Laughing 3a241e4cea
update segment training (#57)
2 years ago
Ayush Chaurasia d0b0fe2592
CLI updates (#58)
2 years ago
Ayush Chaurasia c5f5b80c04
Deterministic training (#53)
2 years ago
Ayush Chaurasia d7df1770fa
Add warmup and accumulation (#52)
2 years ago
Ayush Chaurasia 512a225ce8
Cli support (#50)
2 years ago
Ayush Chaurasia 832ea56eb4
update model initialization design, supports custom data/num_classes (#44)
2 years ago
Ayush Chaurasia f56c9bcc26
Segmentation support & other enchancements (#40)
2 years ago
Glenn Jocher e4f7458d90
Hydra *.yml extension deprecated fix (#34)
2 years ago
Glenn Jocher 6fe8bead35
Classify training cleanup (#33)
2 years ago
Ayush Chaurasia 1054819a59
Add initial model interface (#30)
2 years ago
Ayush Chaurasia c5cb76b356
Metrics and loss structure (#28)
2 years ago
Ayush Chaurasia d0b3c9812b
Trainer + Dataloaders (#27)
2 years ago