483 Commits (8d7490f060a5320010670a387d8c6947257a2091)

Author SHA1 Message Date
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 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 298287298d
Add clearml logging (#51)
2 years ago
Ayush Chaurasia 512a225ce8
Cli support (#50)
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
Glenn Jocher 8530e3fae0
Logging fix from YOLOv5 (#47)
2 years ago
Laughing 47f1cb3ef4
Fix some cuda training issues of segmentation (#46)
2 years ago
Ayush Chaurasia db1031a1a9
Allow setting model attributes before training (#45)
2 years ago
Ayush Chaurasia 832ea56eb4
update model initialization design, supports custom data/num_classes (#44)
2 years ago
Ayush Chaurasia 1f3aad86c1
new check_dataset functions (#43)
2 years ago
Ayush Chaurasia d143ac666f
[rename] - preprocess-batch -> preprocess, preprocess_preds -> postprocess (#42)
2 years ago
Glenn Jocher 4c68b9dcf6
Add is_colab() and is_kaggle() (#41)
2 years ago
Ayush Chaurasia f56c9bcc26
Segmentation support & other enchancements (#40)
2 years ago
Laughing c617ee1c79
Fix dataloader2 (#35)
2 years ago
Glenn Jocher 523eff99e2
Update ci.yaml (#37)
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
Laughing 2e9b18ce4e
Fix dataloader (#32)
2 years ago
Ayush Chaurasia 92c60758dd
Smart Model loading (#31)
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 c5cb76b356
Metrics and loss structure (#28)
2 years ago
Ayush Chaurasia d0b3c9812b
Trainer + Dataloaders (#27)
2 years ago
Glenn Jocher 171ad46d6b
Create __init__.py (#17)
2 years ago
Ayush Chaurasia 419f6d7167
Add yolo package structure (#15)
2 years ago
Glenn Jocher 4aa74bb726
Normalize version with dev0 (#11)
2 years ago
Ayush Chaurasia 39e9f3cae9
Initial pip structure (#1)
2 years ago